def test_check_label_name():
my_schema = {"label": ClassLabel(names=["red", "green", "blue"])}
ds = Dataset("./data/test/dataset2", shape=(5,), mode="w", schema=my_schema)
ds["label", 0] = 1
ds["label", 1] = 2
ds["label", 0] = 1
ds["label", 1] = 2
ds["label", 2] = 0
assert ds.compute(label_name=True).tolist() == [
{"label": "green"},
{"label": "blue"},
{"label": "red"},
{"label": "red"},
{"label": "red"},
]
assert ds.compute().tolist() == [
{"label": 1},
{"label": 2},
{"label": 0},
{"label": 0},
{"label": 0},
]
assert ds[1].compute(label_name=True) == {"label": "blue"}
assert ds[1].compute() == {"label": 2}
assert ds[1:3].compute(label_name=True).tolist() == [
{"label": "blue"},
{"label": "red"},
]
assert ds[1:3].compute().tolist() == [{"label": 2}, {"label": 0}]
def test_dataset_compute():
dt = {
"first": Tensor(shape=(2,)),
"second": "float",
"text": Text(shape=(None,), max_shape=(12,)),
}
url = "./data/test/ds_compute"
ds = Dataset(schema=dt, shape=(2,), url=url, mode="w")
ds["text", 1] = "hello world"
ds["second", 0] = 3.14
ds["first", 0] = np.array([5, 6])
comp = ds.compute()
comp0 = comp[0]
assert (comp0["first"] == np.array([5, 6])).all()
assert comp0["second"] == 3.14
assert comp0["text"] == ""
comp1 = comp[1]
assert (comp1["first"] == np.array([0, 0])).all()
assert comp1["second"] == 0
assert comp1["text"] == "hello world"
