def test_remove_long_seq(): maxlen = 5 seq = [ [1, 2, 3], [1, 2, 3, 4, 5, 6], ] label = ['a', 'b'] new_seq, new_label = sequence._remove_long_seq(maxlen, seq, label) assert new_seq == [[1, 2, 3]] assert new_label == ['a']