Made sure ordering of matrices are ok.
This commit is contained in:
@@ -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()
|
||||
|
Reference in New Issue
Block a user