def test__create_classes_with_primitives(self): """ Test the `_create_classes_with_primitives` method. """ schema = Schema(Mock()) mock_primitive = Mock() schema._create_class_with_premitives = mock_primitive schema._create_classes_with_primitives(list("Test!!")) self.assertEqual(mock_primitive.call_count, 6)
def test_create(self): """ Test the `create` method. """ schema = Schema(Mock()) # mock function calls mock_primitive = Mock() mock_complex = Mock() schema._create_classes_with_primitives = mock_primitive schema._create_complex_properties_from_classes = mock_complex schema.create("test/schema/schema_company.json") # with read from file mock_primitive.assert_called_with(schema_company_local["classes"]) mock_complex.assert_called_with(schema_company_local["classes"])