split check for multiple studies

bihealth · Oct 31, 2023 · 3ddd6bd · 3ddd6bd
1 parent 6ad18e8
commit 3ddd6bd
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/altamisa/isatab/modify_investigation.py b/altamisa/isatab/modify_investigation.py
@@ -19,6 +19,10 @@ def __init__(self, input_path: str):
         with i_file.open("rt") as f:
             self.investigation = InvestigationReader.from_stream(f).read()
 
+        if len(self.investigation.studies) != 1:
+            # TODO: add support for multiple studies
+            raise IndexError("Only single study investigations are supported.")
+
     @staticmethod
     def _join_protocols(x: dict, y: dict) -> dict:
         """
@@ -58,7 +62,7 @@ def add_assay(self, input_path: str):
         with i_file.open("rt") as f:
             investigation2 = InvestigationReader.from_stream(f).read()
 
-        if len(self.investigation.studies) != 1 or len(investigation2.studies) != 1:
+        if len(investigation2.studies) != 1:
             # TODO: add support for multiple studies
             raise IndexError("Only single study investigations are supported.")