def test_serialize_obj(self): data = ModelDataSet(Apple) data.resource_name = "apple" apple = Apple(name="test") apple.save() serialized = data.serialize_obj(apple) self.assertEqual(serialized, {"id": 1, "name": "test", "resource_uri": "/v1/apple/1/"}) serialized = data.serialize_obj(apple, ["name"]) self.assertEqual(serialized, {"id": 1, "name": "test", "resource_uri": "/v1/apple/1/"})