From 164e6939f768340d7c354577ba383da73919411c Mon Sep 17 00:00:00 2001 From: David Rodriguez Date: Sat, 15 Feb 2025 19:27:41 +0000 Subject: [PATCH] Create alias for save method --- astrodbkit/astrodb.py | 3 ++- astrodbkit/tests/test_astrodb.py | 3 ++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/astrodbkit/astrodb.py b/astrodbkit/astrodb.py index 7b89545..a723042 100644 --- a/astrodbkit/astrodb.py +++ b/astrodbkit/astrodb.py @@ -358,8 +358,9 @@ def __init__( connection_string, sqlite_foreign=sqlite_foreign, connection_arguments=connection_arguments ) - # Convenience methods + # Convenience methods and aliases self.query = self.session.query + self.save = self.save_database self.save_db = self.save_database self.load_db = self.load_database diff --git a/astrodbkit/tests/test_astrodb.py b/astrodbkit/tests/test_astrodb.py index a17d8f1..ca8db36 100644 --- a/astrodbkit/tests/test_astrodb.py +++ b/astrodbkit/tests/test_astrodb.py @@ -450,7 +450,8 @@ def test_save_database(db, db_dir): elif os.path.isdir(file_path): shutil.rmtree(file_path) - db.save_database(db_dir) + # Can use db.save, db.save_db, or db.save_database + db.save(db_dir) # Check JSON data assert os.path.exists(os.path.join(db_dir, "reference", 'Publications.json'))