class TestFileReader(unittest.TestCase): def setUp(self): self.temp = tempfile.NamedTemporaryFile('w+t') self.fr = FileReader('pattern','negative') def test_search(self): self.temp.write('line1 pattern\n') self.temp.write('line2 pattern negative\n') self.temp.write('line3 negative\n') self.temp.write('line4\n') self.temp.flush() self.temp.seek(0) data = self.fr.process_file(self.temp) self.assertEquals(len(data), 1) self.assertTrue('line1' in data[0]['text']) def tearDown(self): self.temp.close()
def setUp(self): self.temp = tempfile.NamedTemporaryFile('w+t') self.fr = FileReader('pattern','negative')