diff options
Diffstat (limited to 'verif/tests/test_tosa_datagenerator.py')
-rw-r--r-- | verif/tests/test_tosa_datagenerator.py | 14 |
1 files changed, 14 insertions, 0 deletions
diff --git a/verif/tests/test_tosa_datagenerator.py b/verif/tests/test_tosa_datagenerator.py index ba0235c..4f3d7fd 100644 --- a/verif/tests/test_tosa_datagenerator.py +++ b/verif/tests/test_tosa_datagenerator.py @@ -114,3 +114,17 @@ def test_generate_dot_product_check_fail_names(): for f in json_config["ifm_file"]: file = TEST_DIR / f assert not file.is_file() + + +@pytest.mark.postcommit +def test_generate_tensor_data_check(): + glib = GenerateLibrary(GENERATE_LIB_PATH) + assert glib + + json_config = JSON_DATAGEN_DOT_PRODUCT["meta"]["data_gen"] + + for n in JSON_DATAGEN_DOT_PRODUCT["ifm_name"]: + arr = glib.get_tensor_data(n, json_config) + + assert arr.shape == tuple(json_config["tensors"][n]["shape"]) + assert arr.dtype == np.float32 |