def test_to_vw(self): stream = TextFileStreamer(path_list=[self.doc1, self.doc2], tokenizer=self.tokenizer) result = StringIO() stream.to_vw(result) benchmark = " 1 doc1| failure:1 doomed:1\n 1 doc2| set:1 success:1\n" self.assertEqual(benchmark, result.getvalue())