mongoengine/tests/test_common.py
2019-10-31 22:39:53 +01:00

15 lines
393 B
Python

import unittest
from mongoengine import Document
from mongoengine.common import _import_class
class TestCommon(unittest.TestCase):
def test__import_class(self):
doc_cls = _import_class("Document")
self.assertIs(doc_cls, Document)
def test__import_class_raise_if_not_known(self):
with self.assertRaises(ValueError):
_import_class("UnknownClass")