Merge pull request #10 from Australian-Structural-Biology-Computing/p…

…atch-yaml Patch YAML
Australian-Structural-Biology-Computing · Dec 3, 2024 · 9787c1b · 9787c1b
2 parents 1a58825 + 09fd530
commit 9787c1b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/create_samplesheet.py b/create_samplesheet.py
@@ -85,10 +85,10 @@ def create_yaml(data, fp):
         "sequences": []
     }
     for row in data:
-        output_data["sequences"].append({
+        output_data["sequences"].append({"protein": {
             "id": row.name, 
             "sequence": row.data
-        })
+        }})
 
     yaml.dump(output_data, fp, default_flow_style=False)
 

diff --git a/test.py b/test.py
@@ -265,7 +265,7 @@ def test_create_yaml_single_sample(self):
         fp_data = fp.read()
         fp.close()
 
-        self.assertEqual(fp_data, 'sequences:\n- id: TEST\n  sequence: AAAAAA\nversion: 1\n')
+        self.assertEqual(fp_data, 'sequences:\n- protein:\n    id: TEST\n    sequence: AAAAAA\nversion: 1\n')
 
     def test_create_yaml_multiple_sample(self):
         sample_input = []
@@ -280,7 +280,7 @@ def test_create_yaml_multiple_sample(self):
         fp_data = fp.read()
         fp.close()
 
-        self.assertEqual(fp_data, "sequences:\n- id: TEST1\n  sequence: AAAAAA\n- id: TEST2\n  sequence: AAAAAA\n- id: TEST3\n  sequence: AAAAAA\nversion: 1\n")
+        self.assertEqual(fp_data, "sequences:\n- protein:\n    id: TEST1\n    sequence: AAAAAA\n- protein:\n    id: TEST2\n    sequence: AAAAAA\n- protein:\n    id: TEST3\n    sequence: AAAAAA\nversion: 1\n")
 
 if __name__ == "__main__":
     unittest.main()