def test_instantiation(self): """Test proper instantiation of object 'User()'""" am = Amenity() self.assertIsInstance(am, Amenity) self.assertIsInstance(am.id, str) self.assertIsInstance(am.created_at, datetime.datetime) self.assertIsInstance(am.updated_at, datetime.datetime) self.assertIsInstance(am.__class__, type) am.size = "tall" l1 = dir(am) self.assertIn('size', l1) self.assertEqual(am.__dict__['size'], 'tall') am.size = 'tall' l2 = dir(am) self.assertIn('size', l2) self.assertEqual(am.__dict__['size'], 'tall') am.age = 28 l3 = dir(am) self.assertIn('age', l3) self.assertEqual(am.__dict__['age'], 28) am.age = 28.5 l4 = dir(am) self.assertIn('age', l4) self.assertEqual(am.__dict__['age'], 28.5) am.age = None l5 = dir(am) self.assertIn('age', l5) self.assertEqual(am.__dict__['age'], None) am_kw1 = Amenity(**{}) self.assertIsInstance(am_kw1, Amenity) self.assertIsInstance(am_kw1.id, str) self.assertIsInstance(am_kw1.created_at, datetime.datetime) self.assertIsInstance(am_kw1.updated_at, datetime.datetime) self.assertIsInstance(am_kw1.__class__, type) am_kw2 = Amenity(**{"first_name": "John", "age": 25}) l6 = dir(am_kw2) self.assertIn('first_name', l6) self.assertIn('age', l6) self.assertEqual(am_kw2.__dict__['first_name'], 'John') self.assertEqual(am_kw2.__dict__['age'], 25)
def test_to_dict(self): """Test to_dict method""" am = Amenity() am.age = 28 am.size = "tall" for k, v in am.__dict__.items(): if k != 'updated_at' and k != 'created_at': self.assertIn(k, am.to_dict()) self.assertEqual(v, am.to_dict()[k]) self.assertEqual(am.to_dict()['__class__'], am.__class__.__name__) self.assertEqual(am.to_dict()['updated_at'], am.updated_at.isoformat()) self.assertEqual(am.to_dict()['created_at'], am.created_at.isoformat()) self.assertEqual(am.to_dict()['age'], 28) self.assertEqual(am.to_dict()['size'], 'tall') self.assertIsInstance(am.to_dict(), dict)