class ModelTest(unittest.TestCase): def setUp(self): self.model = ModelBase('foo', {'foo': 'bar', 'here': 1}) def test_data_fields_returns_list_of_attribute_keys(self): data_fields = self.model.data_fields() six.assertCountEqual(self, data_fields, ['foo', 'here']) def test_data_can_be_accessed_as_dict(self): self.assertEqual(self.model['foo'], 'bar') self.assertEqual(self.model['here'], 1) def test_data_can_be_accessed_by_attribute_name(self): self.assertEqual(self.model.foo, 'bar') self.assertEqual(self.model.here, 1) def test_throws_exception_when_attribute_does_not_exist(self): self.assertRaises(AttributeError, lambda: self.model.blah) def test_to_list_returns_values(self): results = self.model.to_list() six.assertCountEqual(self, results, ['bar', 1])
def setUp(self): self.model = ModelBase('foo', {'foo': 'bar', 'here': 1})