예제 #1
0
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)
예제 #2
0
    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)