def testLoadPeople(self):
#Shouldn't work when wrong delimiter is specified
thrown = False
try:
person = Person.loadPeople("tests/_td/dosage_set_1/set.sample",
delim=",")
except:
thrown = True
self.assertEqual(thrown, True)
#default load
people = Person.loadPeople("tests/_td/dosage_set_1/set.sample")
self.assertIsNotNone(people)
self.assertEqual(len(people), 5)
self.assertPerson(people[0], "ID1", "K", "HERO", "male")
self.assertPerson(people[1], "ID2", "K", "HERO", "female")
self.assertPerson(people[2], "DI5", "K", "HERO", "male")
self.assertPerson(people[3], "ID3", "K", "HERO", "female")
self.assertPerson(people[4], "B1", "L", "T", "female")
#let's pretend the file header is a person entry
people = Person.loadPeople("tests/_td/dosage_set_1/set.sample",
skip_header=False)
self.assertIsNotNone(people)
self.assertEqual(len(people), 6)
self.assertPerson(people[0], "ID", "POP", "GROUP", "SEX")
self.assertPerson(people[1], "ID1", "K", "HERO", "male")
self.assertPerson(people[2], "ID2", "K", "HERO", "female")
self.assertPerson(people[3], "DI5", "K", "HERO", "male")
self.assertPerson(people[4], "ID3", "K", "HERO", "female")
self.assertPerson(people[5], "B1", "L", "T", "female")
def testLoadPeople(self):
# Shouldn't work when wrong delimiter is specified
thrown = False
try:
person = Person.loadPeople("tests/_td/dosage_set_1/set.sample", delim=",")
except:
thrown = True
self.assertEqual(thrown, True)
# default load
people = Person.loadPeople("tests/_td/dosage_set_1/set.sample")
self.assertIsNotNone(people)
self.assertEqual(len(people), 5)
self.assertPerson(people[0], "ID1", "K", "HERO", "male")
self.assertPerson(people[1], "ID2", "K", "HERO", "female")
self.assertPerson(people[2], "DI5", "K", "HERO", "male")
self.assertPerson(people[3], "ID3", "K", "HERO", "female")
self.assertPerson(people[4], "B1", "L", "T", "female")
# let's pretend the file header is a person entry
people = Person.loadPeople("tests/_td/dosage_set_1/set.sample", skip_header=False)
self.assertIsNotNone(people)
self.assertEqual(len(people), 6)
self.assertPerson(people[0], "ID", "POP", "GROUP", "SEX")
self.assertPerson(people[1], "ID1", "K", "HERO", "male")
self.assertPerson(people[2], "ID2", "K", "HERO", "female")
self.assertPerson(people[3], "DI5", "K", "HERO", "male")
self.assertPerson(people[4], "ID3", "K", "HERO", "female")
self.assertPerson(people[5], "B1", "L", "T", "female")
