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"