Example #1
0
    def test_callback(self):
        vect = SimhashVectorizer(shingle_len=10, f=64)
        callback = MagicMock()
        result = vect.transform(self.corpus, callback=callback)

        self.assertIsInstance(result, Corpus)
        self.assertEqual(len(result), len(self.corpus))
        self.assertEqual(result.X.shape, (len(self.corpus), 64))
        callback.assert_has_calls(
            [call(i / len(self.corpus)) for i in range(9)])
Example #2
0
 def test_report(self):
     vect = SimhashVectorizer()
     self.assertGreater(len(vect.report()), 0)
Example #3
0
 def test_report(self):
     vect = SimhashVectorizer()
     self.assertGreater(len(vect.report()), 0)
Example #4
0
 def test_transform(self):
     vect = SimhashVectorizer(shingle_len=10, f=64)
     result = vect.transform(self.corpus)
     self.assertIsInstance(result, Corpus)
     self.assertEqual(len(result), len(self.corpus))
     self.assertEqual(result.X.shape, (len(self.corpus), 64))
Example #5
0
 def test_transform(self):
     vect = SimhashVectorizer(shingle_len=10, f=64)
     result = vect.transform(self.corpus)
     self.assertIsInstance(result, Corpus)
     self.assertEqual(len(result), len(self.corpus))
     self.assertEqual(result.X.shape, (len(self.corpus), 64))