16 lines
394 B
Python
16 lines
394 B
Python
import unittest
|
|
|
|
from mongoengine.common import _import_class
|
|
from mongoengine import Document
|
|
|
|
|
|
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")
|