Added categorygetter for PhenotypeDataset.

2006-05-04 13:30:04 +00:00
parent 338778d7b1
commit fafa373e67
2 changed files with 22 additions and 2 deletions
--- a/test/workflows/affy_workflowtest.py
+++ b/test/workflows/affy_workflowtest.py
@@ -75,6 +75,17 @@ CEL\tsex\tage\tinfected
                           ['02-05-34', 'F', '9', 'N'],
                           ['02-05-35', 'M', '8', 'I']], dataset.get_phenotype_table())

+    def testGetCategories(self):
+        cel_data = """\
+CEL\tsex\tage\tinfected
+02-05-33\tF\t8\tI
+02-05-34\tF\t9\tN
+02-05-35\tM\t8\tI
+"""
+        dataset = PhenotypeDataset(cel_data)
+
+        self.assertEquals(set(['F', 'M', 'I', 'N']), set(dataset.get_categories()))
+


 if __name__=='__main__':