def test_iter(self):
"""Prueba el iterador
"""
self.assertSequenceEqual(sorted(iter(BagOfWords())), [])
self.assertSequenceEqual(sorted(iter(BagOfWords("cat cow dog"))), [("cat", 1), ("cow", 1), ("dog", 1)])
self.assertSequenceEqual(sorted(iter(BagOfWords(text="cat dog cat"))), [("cat", 2), ("dog", 1)])
def test_len(self):
"""Prueba el tamaño del vector
"""
self.assertEqual(len(BagOfWords()), 0)
self.assertEqual(len(BagOfWords("cat dog cow")), 3)
self.assertEqual(len(BagOfWords(text="cat dog cat")), 2)
def test_init_with_dict(self):
"""Prueba la inicialización con diccinoarios
"""
self.assertDictEqual(BagOfWords(values={"cat": 1, "dog": 1, "cow": 1}).values, {"cat": 1, "dog": 1, "cow": 1})
self.assertDictEqual(BagOfWords(values={"cat": 2, "dog": 1}).values, {"cat": 2, "dog": 1})
def test_init_with_symbols_in_str(self):
"""Prueba la inicialización con strings que contengan símbolos de puntuación
"""
self.assertDictEqual(BagOfWords(" cat, dog! cow.").values, {"cat": 1, "dog": 1, "cow": 1})
self.assertDictEqual(BagOfWords(text="cat dog?? cat!!! ").values, {"cat": 2, "dog": 1})
def test_init_with_str(self):
"""Prueba la inicialización con strings
"""
self.assertDictEqual(BagOfWords("cat dog cow").values, {"cat": 1, "dog": 1, "cow": 1})
self.assertDictEqual(BagOfWords(text="Cat dog cat").values, {"cat": 2, "dog": 1})