Made sure ordering of matrices are ok.

This commit is contained in:
2006-05-09 14:13:09 +00:00
parent 86f89b6ffe
commit 443b18cea6
2 changed files with 50 additions and 17 deletions

View File

@@ -75,6 +75,13 @@ CEL\tsex\tage\tinfected
['02-05-34', 'F', '9', 'N'],
['02-05-35', 'M', '8', 'I']], dataset.get_phenotype_table())
# we can also get a sorted list
new_order = ['02-05-35', '02-05-33', '02-05-34']
self.assertEquals([['CEL', 'sex', 'age', 'infected'],
['02-05-35', 'M', '8', 'I'],
['02-05-33', 'F', '8', 'I'],
['02-05-34', 'F', '9', 'N']], dataset.get_phenotype_table(new_order))
def testGetCategories(self):
cel_data = """\
CEL\tsex\tage\tinfected
@@ -110,6 +117,5 @@ CEL\tsex\tage\tinfected
self.assertEquals([1, 0, 1], dataset.get_category_variable("I"))
self.assertEquals([0, 1, 0], dataset.get_category_variable("N"))
if __name__=='__main__':
unittest.main()