update dataset/series registry

cbyrohl · Aug 17, 2023 · 429c44f · 429c44f
1 parent e6da552
commit 429c44f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/src/scida/__init__.py b/src/scida/__init__.py
@@ -13,5 +13,6 @@
 from scida.customs.gizmo.series import GizmoSimulation
 from scida.customs.rockstar.dataset import RockstarCatalog
 from scida.customs.swift.dataset import SwiftSnapshot
+from scida.customs.swift.series import SwiftSimulation
 
 logging.basicConfig(stream=sys.stdout, level=logging.INFO)
diff --git a/tests/customs/test_flamingo.py b/tests/customs/test_flamingo.py
@@ -63,6 +63,8 @@ def test_snapshot(testdatapath):
 def test_series(testdatapath):
     obj: SwiftSimulation = load(testdatapath, units=True)
     obj.info()
+    assert isinstance(obj, SwiftSimulation)
     z = 4.75
     ds = obj.get_dataset(redshift=z)
+    assert isinstance(ds, SwiftSnapshot)
     assert ds.redshift == z