def test_flatten_with_sequence(self):
features = Features(
{"foo": Sequence({"bar": {
"my_value": Value("int32")
}})})
_features = features.copy()
flattened_features = features.flatten()
assert flattened_features == {
"foo.bar": [{
"my_value": Value("int32")
}]
}
assert features == _features, "calling flatten shouldn't alter the current features"
def test_flatten(self):
features = Features({
"foo": {
"bar1": Value("int32"),
"bar2": {
"foobar": Value("string")
}
}
})
_features = features.copy()
flattened_features = features.flatten()
assert flattened_features == {
"foo.bar1": Value("int32"),
"foo.bar2.foobar": Value("string")
}
assert features == _features, "calling flatten shouldn't alter the current features"
