def test_valid_passport(self): bela = Donor() bela.uniqueid = 'ASDFGH12' self.assertEqual(bela.type_of_doc(), 'PASSPORT')
def test_valid_ID(self): bela = Donor() bela.uniqueid = '123456AB' self.assertEqual(bela.type_of_doc(), 'ID')