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