mongoengine/tests/test_common.py

17 lines
379 B
Python

import unittest
import pytest
from mongoengine import Document
from mongoengine.common import _import_class
class TestCommon:
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")