class PersonTestCase(unittest.TestCase): def setUp(self): pre_populate.populate() self.me = Person('john', 'doe') def tearDown(self): del self.me reset.reset() def test_class(self): '''test if object is of class Person''' self.assertTrue(isinstance(self.me, Person), 'object is not an instance of Person') def test_names(self): '''test first and last names set correctly''' self.assertEqual(self.me.first_name, 'john', 'object first name incorrect') self.assertEqual(self.me.last_name, 'doe', 'object last name incorrect') def test_office_room(self): ''' test office room initially set to None''' self.assertIsNone(self.me.office_room, 'office room not initially set to None') def test_assign_office(self): '''test person can be assigned an office''' self.me.assign_office('Staff') self.assertIsNotNone(self.me.office_room, 'office room cannot be assigned to person') def test_id(self): '''test Person object has no id''' self.assertIsNone(self.me.id, 'An Object of Person cannot have an id')
def setUp(self): pre_populate.populate() self.me = Person('john', 'doe')