def test_right_padded_sequence_assertion(): right_padded_sequence = [[True, True, False, False], [True, True, True, False]] left_padded_sequence = [[False, False, True, True], [False, True, True, True]] _ = sampler_py._check_sequence_is_right_padded(right_padded_sequence, False) with pytest.raises(tf.errors.InvalidArgumentError): _ = sampler_py._check_sequence_is_right_padded(left_padded_sequence, False)
def testRightPaddedSequenceAssertion(self): right_padded_sequence = [[True, True, False, False], [True, True, True, False]] left_padded_sequence = [[False, False, True, True], [False, True, True, True]] assertion = sampler_py._check_sequence_is_right_padded( right_padded_sequence, False ) self.evaluate(assertion) with self.assertRaises(tf.errors.InvalidArgumentError): assertion = sampler_py._check_sequence_is_right_padded( left_padded_sequence, False ) self.evaluate(assertion)