def test_truncate_to_length_no_arg(self):
    """Tests that a no-arg call leaves shapes unchanged."""
    def data_stream():
      while True:
        yield (np.zeros((1, 5)), np.ones((1, 5)))
    stream_fn = data.inputs.TruncateToLength()
    y0, y1 = next(stream_fn(data_stream()))
    self.assertEqual(y0.shape, (1, 5))
    self.assertEqual(y1.shape, (1, 5))
Tests that a no-arg call leaves shapes unchanged.
trax/data/inputs_test.py