def test_GIVEN_string_list_WHEN_getting_data_and_type_THEN_returns_correct_dtype( file): dataset_name = "ds" dataset_value = np.string_(["s", "t", "r"]) dataset = file.create_dataset(dataset_name, data=dataset_value) data, dtype, size = get_data_and_type(dataset) assert data == [x.decode("ASCII") for x in list(dataset_value)] assert size == (len(dataset_value), )
def test_GIVEN_array_WHEN_getting_data_and_dtype_THEN_function_returns_correcte_fw_json_dtype_and_values( file, ): expected_dtype = "float" expected_values = [1.1, 1.2, 1.3] dataset = file.create_dataset("test_dataset", data=expected_values, dtype="float32") data, dtype, size = get_data_and_type(dataset) assert size == (len(expected_values), ) assert np.allclose(data, expected_values) assert dtype == expected_dtype
def test_GIVEN_float_WHEN_getting_data_and_type_THEN_returns_correct_dtype( file): dataset_name = "ds" dataset_type = np.float32 dataset_value = np.float32(2.123) dataset = file.create_dataset(dataset_name, dtype=dataset_type, data=dataset_value) data, dtype, size = get_data_and_type(dataset) assert data == dataset_value assert dtype == "float" assert size == 1
def test_GIVEN_single_string_WHEN_getting_data_and_dtype_THEN_function_returns_correct_fw_json_dtype( file, ): expected_dtype = "string" expected_size = 1 expected_value = np.string_("udder") dataset = file.create_dataset("test_dataset", data=expected_value, dtype="S5") data, dtype, size = get_data_and_type(dataset) assert size == expected_size assert dtype == expected_dtype assert bytes(data, "ASCII") == expected_value
def test_GIVEN_int64_WHEN_getting_data_and_dtype_THEN_function_returns_correct_fw_json_dtype( file, ): expected_dtype = "int64" expected_size = 1 expected_value = np.int64(171_798_691_842) # bigger than max 32b int dataset = file.create_dataset("test_dataset", dtype="int64", data=expected_value) data, dtype, size = get_data_and_type(dataset) assert size == expected_size assert dtype == expected_dtype assert data == expected_value
def test_GIVEN_int32_WHEN_getting_data_and_dtype_THEN_function_returns_correct_fw_json_dtype( file, ): expected_dtype = "int32" expected_size = 1 expected_value = np.int32(42) dataset = file.create_dataset("test_dataset", dtype="int32", data=expected_value) data, dtype, size = get_data_and_type(dataset) assert size == expected_size assert dtype == expected_dtype assert data == expected_value
def test_GIVEN_float64_WHEN_getting_data_and_dtype_THEN_function_returns_correct_fw_json_dtype( file, ): expected_dtype = "double" expected_size = 1 expected_value = np.float64(324.123_231_413_515_223_412_352_135_34) dataset = file.create_dataset("test_dataset", dtype=np.float64, data=expected_value) data, dtype, size = get_data_and_type(dataset) assert size == expected_size assert dtype == expected_dtype assert data == expected_value