def test_round_trip_int_list_feature(self):
example = tf.train.SequenceExample()
key = "test_ints_features"
msu.add_int_list(key, [0, 1], example)
msu.add_int_list(key, [1, 2], example)
self.assertEqual(2, msu.get_feature_list_size(key, example))
self.assertTrue(msu.has_feature_list(key, example))
self.assertAllClose([0, 1], msu.get_int_list_at(key, 0, example))
self.assertAllClose([1, 2], msu.get_int_list_at(key, 1, example))
msu.clear_feature_list(key, example)
self.assertEqual(0, msu.get_feature_list_size(key, example))
self.assertFalse(msu.has_feature_list(key, example))
def test_round_trip_bytes_list_feature(self):
example = tf.train.SequenceExample()
key = "test_bytes_features"
msu.add_bytes_list(key, [b"test0", b"test1"], example)
msu.add_bytes_list(key, [b"test1", b"test2"], example)
self.assertEqual(2, msu.get_feature_list_size(key, example))
self.assertTrue(msu.has_feature_list(key, example))
self.assertAllEqual([b"test0", b"test1"],
msu.get_bytes_list_at(key, 0, example))
self.assertAllEqual([b"test1", b"test2"],
msu.get_bytes_list_at(key, 1, example))
msu.clear_feature_list(key, example)
self.assertEqual(0, msu.get_feature_list_size(key, example))
self.assertFalse(msu.has_feature_list(key, example))