def test_reverse_transform_string(self):
     vectorizer = CharVectorizer("abc", map_unknown_chars_to="X")
     texts = ["aaa", "bbb", "ccc", "abc", "???", "a?a"]
     expected = ["aaa", "bbb", "ccc", "abc", "XXX", "aXa"]
     
     matrices = []
     for text in texts:
         matrices.append(vectorizer.transform_string(text, len(texts[0])))
     
     reverse_transformed = []
     for matrix in matrices:
         for row in matrix:
             reverse_transformed.append(vectorizer.reverse_transform_string(row))
     
     for text_is, text_exp in zip(reverse_transformed, expected):
         self.assertEqual(text_is, text_exp)
Beispiel #2
0
    def test_reverse_transform_string(self):
        vectorizer = CharVectorizer("abc", map_unknown_chars_to="X")
        texts = ["aaa", "bbb", "ccc", "abc", "???", "a?a"]
        expected = ["aaa", "bbb", "ccc", "abc", "XXX", "aXa"]

        matrices = []
        for text in texts:
            matrices.append(vectorizer.transform_string(text, len(texts[0])))

        reverse_transformed = []
        for matrix in matrices:
            for row in matrix:
                reverse_transformed.append(
                    vectorizer.reverse_transform_string(row))

        for text_is, text_exp in zip(reverse_transformed, expected):
            self.assertEqual(text_is, text_exp)