View source on GitHub |
Test item of a FeatureExpectation.
tfds.testing.FeatureExpectationItem(
value: Any,
expected: Optional[Any] = None,
expected_np: Optional[np.ndarray] = None,
expected_serialized: Optional[Any] = None,
decoders: Optional[utils.TreeDict[Any]] = None,
dtype: Optional[tf.dtypes.DType] = None,
shape: Optional[utils.Shape] = None,
raise_cls: Optional[Type[Exception]] = None,
raise_cls_np: Optional[Type[Exception]] = None,
raise_msg: Optional[str] = None,
atol: Optional[float] = None
)
Should be passed to assertFeature
method of FeatureExpectationsTestCase
.
Each FeatureExpectationItem
test an example serialization/deserialization (
feature.encode_example
-> example_serializer.serialize_example
->
example_parse.parse_example
-> feature.decode_example
).
Attributes | |
---|---|
value
|
Input to features.encode_example
|
expected
|
Expected output after features.decode_example
|
expected_np
|
Expected output after features.decode_example_np
|
expected_serialized
|
Optional |
decoders
|
Optional tfds.decode.Decoder objects (to overwrite the default
features.decode_example ). See
https://www.tensorflow.org/datasets/decode.
|
dtype
|
If decoders is provided, the output of decode_example is checked
against this value (otherwise, output is checked against features.dtype )
|
shape
|
If decoders is provided, the output of decode_example is checked
against this value (otherwise, output is checked against features.shape )
|
raise_cls
|
Expected error raised during features.encode_example . When set
expected and expected_serialized should be None .
|
raise_cls_np
|
Expected error raised during features.encode_example_np .
When set expected_np and expected_serialized_np should be None .
|
raise_msg
|
Expected error message regex. |
atol
|
If provided, compare float values with this precision (use default otherwise). |
Methods
__eq__
__eq__(
other
)
Class Variables | |
---|---|
atol |
None
|
decoders |
None
|
dtype |
None
|
expected |
None
|
expected_np |
None
|
expected_serialized |
None
|
raise_cls |
None
|
raise_cls_np |
None
|
raise_msg |
None
|
shape |
None
|