Add test case for namespace_match.
@@ -6,3 +6,15 @@ from allennlp.common.testing import AllenNlpTestCase
class TestCommonUtils(AllenNlpTestCase):
def test_group_by_count(self):
assert util.group_by_count([1, 2, 3, 4, 5, 6, 7], 3, 20) == [[1, 2, 3], [4, 5, 6], [7, 20, 20]]
+
+ def test_pad_sequence_to_length(self):
+ assert util.pad_sequence_to_length([1, 2, 3], 5) == [1, 2, 3, 0, 0]
+ assert util.pad_sequence_to_length([1, 2, 3], 5, default_value=lambda: 2) == [1, 2, 3, 2, 2]
+ assert util.pad_sequence_to_length([1, 2, 3], 5, padding_on_right=False) == [0, 0, 1, 2, 3]
+
+ def test_namespace_match(self):
+ assert util.namespace_match("*tags", "tags")
+ assert util.namespace_match("*tags", "passage_tags")
+ assert util.namespace_match("*tags", "question_tags")
+ assert util.namespace_match("tokens", "tokens")
+ assert not util.namespace_match("tokens", "stemmed_tokens")
