mongoengine/tests/test_common.py
2019-10-31 22:59:49 +01:00

17 lines
398 B
Python

import unittest
import pytest
from mongoengine import Document
from mongoengine.common import _import_class
class TestCommon(unittest.TestCase):
def test__import_class(self):
doc_cls = _import_class("Document")
assert doc_cls is Document
def test__import_class_raise_if_not_known(self):
with pytest.raises(ValueError):
_import_class("UnknownClass")