Python MongoStreamer.to_scipysparse Examples

Example #1

File: test_streamer.py Project: vickingur/rosetta

    def test_to_scipyspare(self):
        stream = MongoStreamer(self.db_setup, tokenizer=self.tokenizer)

        stream.cursor = self.mock_cursor
        result = stream.to_scipysparse()
        benchmark = sparse.csr_matrix([[1, 1, 0, 0], [0, 0, 1, 1]])

        compare = result.toarray() == benchmark.toarray()
        self.assertTrue(compare.all())

Example #2

File: test_streamer.py Project: rjweiss/rosetta

    def test_to_scipyspare(self):
        stream = MongoStreamer(self.db_setup, tokenizer=self.tokenizer)

        stream.cursor = self.mock_cursor
        result = stream.to_scipysparse()
        benchmark = sparse.csr_matrix([[1, 1, 0, 0], [0, 0, 1, 1]])

        compare = result.toarray() == benchmark.toarray()
        self.assertTrue(compare.all())