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")