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'))