def test_get_number_of_examples_raises_value_error(model_data: RasaModelData):
model_data.data["dense"] = {}
model_data.data["dense"]["data"] = [np.random.randint(5, size=(2, 10))]
with pytest.raises(ValueError):
model_data.number_of_examples()
def test_get_number_of_examples(model_data: RasaModelData):
assert model_data.number_of_examples() == 5
