def test_get(self): data = ModelDataSet(Apple) apple = Apple(name="test") apple.save() check = data.get(id=1) self.assertEqual(apple, check)