Skip to content

Commit 9393be8

Browse files
Add test for model_type
1 parent c9d6323 commit 9393be8

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tests/test_maker_utils.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -109,6 +109,8 @@ def test_datamodel_maker(model_class):
109109
assert isinstance(model, model_class)
110110
model.validate()
111111

112+
assert model.meta.model_type == model_class.__name__
113+
112114

113115
@pytest.mark.parametrize("node_class", [node for node in datamodels.MODEL_REGISTRY])
114116
@pytest.mark.filterwarnings("ignore:This function assumes shape is 2D")

0 commit comments

Comments
 (0)