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
