Create alias for save method (#98)

astrodbtoolkit · Feb 16, 2025 · 24a4749 · 24a4749
1 parent 0112ab7
commit 24a4749
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git 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
@@ -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'))