Very small fix

Author: AstroPatty

opencosmo/link/collection.py

@@ -58,7 +58,7 @@ def __init__(
         self.__filters = filters
 
     def __repr__(self):
-        structure_type = self.header.file.data_type.split("_")[0] + "s"
+        structure_type = self.__header.file.data_type.split("_")[0] + "s"
         dtype_str = ", ".join(self.__handlers.keys())
         return f"Collection of {structure_type} with linked datasets {dtype_str}"
 

opencosmo/link/io.py

@@ -92,6 +92,10 @@ def open_linked_files(*files: Path):
     Open a collection of files that are linked together, such as a
     properties file and a particle file.
     """
+    if len(files) == 1 and isinstance(files[0], list):
+        return open_linked_files(*files[0])
+
+
     file_handles = [File(file, "r") for file in files]
     headers = [read_header(file) for file in file_handles]
     properties_file, linked_files = verify_links(*headers)

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "opencosmo"
-version = "0.6.0"
+version = "0.6.1"
 description = "OpenCosmo Python Toolkit"
 authors = ["Patrick Wells <pwells@anl.gov>"]
 readme = "README.md"

test/test_collection.py

@@ -42,6 +42,9 @@ def test_multi_filter(multi_path):
     for ds in collection.values():
         assert all(ds.data["sod_halo_mass"] > 0)
 
+def test_multi_repr(multi_path):
+    collection = oc.read(multi_path)
+    assert isinstance(collection.__repr__(), str)
 
 def test_multi_filter_write(multi_path, tmp_path):
     collection = oc.read(multi_path)
@@ -85,6 +88,11 @@ def test_data_linking(halo_paths):
     assert n_particles > 0
     assert n_profiles > 0
 
+def test_data_link_repr(halo_paths):
+    collection = open_linked_files(halo_paths)
+    assert isinstance(collection.__repr__(), str)
+
+
 
 def test_data_link_selection(halo_paths):
     collection = open_linked_files(*halo_paths)

test/test_read.py

@@ -34,3 +34,7 @@ def test_all_columns_included(input_path):
 def test_read_multi(multi_path):
     dataset = oc.read(multi_path)
     assert isinstance(dataset, oc.collection.collection.SimulationCollection)
+
+def test_dataset_repr(input_path):
+    dataset = oc.read(input_path)
+    assert isinstance(dataset.__repr__(), str)