remove some test code debugging print statements

[htsworkflow.git] / htsworkflow / pipelines / test / test_retrive_config.py

diff --git a/htsworkflow/pipelines/test/test_retrive_config.py b/htsworkflow/pipelines/test/test_retrive_config.py
index cb5650191e79f886da243c9837bcfa61291fbb1f..d06508e9ed3217aa8475f9ace27c39074d0a0202 100644 (file)
--- a/htsworkflow/pipelines/test/test_retrive_config.py
+++ b/htsworkflow/pipelines/test/test_retrive_config.py
@@ -54,19 +54,32 @@ class RetrieveTestCases(TestCase):
 
         output = StringIO()
         save_sample_sheet(output, options, flowcell_info)
+
         output.seek(0)
         sheet = list(csv.DictReader(output))
-        expected = [{'SampleProject': '12044_index1', 'Index': 'ATCACG'},
-                    {'SampleProject': '12044_index2', 'Index': 'CGATGT'},
-                    {'SampleProject': '12044_index3', 'Index': 'TTAGGC'},
-                    {'SampleProject': '11045_index1', 'Index': 'ATCACG'},
+        expected = [{'SampleProject': '12044_index1',
+                     'Index': 'ATCACG',
+                     'Lane': '3',
+                     },
+                    {'SampleProject': '12044_index2',
+                     'Index': 'CGATGT',
+                     'Lane': '3',
+                     },
+                    {'SampleProject': '12044_index3',
+                     'Index': 'TTAGGC',
+                     'Lane': '3',
+                     },
+                    {'SampleProject': '11045_index1',
+                     'Index': 'ATCACG',
+                     'Lane': '3',
+                     },
+                    {'SampleProject': '13044_indexN701-N501',
+                     'Index': 'TAAGGCGA-TAGATCGC',
+                     'Lane': '4',
+                     }
                     ]
-        for i in range(4):
-            self.assertEqual(sheet[i]['SampleProject'],
-                             expected[i]['SampleProject'])
-            self.assertEqual(sheet[i]['Index'],
-                             expected[i]['Index'])
-            self.assertEqual(sheet[i]['FCID'], fcid)
-            self.assertEqual(sheet[i]['Lane'], '3')
-
-
+        self.failUnlessEqual(len(sheet), len(expected))
+        for s, e in zip(sheet, expected):
+            for key in e.keys():
+                self.failUnlessEqual(s[key], e[key],
+                  "%s != %s for key %s" % (s[key],e[key], key))