如何在django中测试我的自定义模型管理器

Question

以下是我的自定义经理

class DefaultModelManager(models.Manager):
def get_or_none(self, **kwargs):
    try:
        return self.get(**kwargs)
    except self.model.DoesNotExist:
        return None

模型

class TestModel(models.Model):
    objects=DefaultModelManager()

我该如何测试这段代码？ 以下是我的代码

class ModelManagerTest(TestCase):
    def test_can_get_or_none(self):
        TestModel.create('...')
        test = TestModel.objects.get_or_none('...')
        test2 = TestModel.objects.get('...')
        self.assertEqual(test, test2)
    def test_cant_get_or_none(self):
        test = TestModel.objects.get_or_none('...')
        self.assertEqual(test, None)

是正确的??? 或其他方法......测试自定义模型管理器的最佳方法是什么