def test_reduce(self): sys.stdin = StringIO( "abc\t1\nabc\t1\nxyz\t1\nabd\t1\nad\t1\nad\t1\nefr\t1") self.reducer.reduce() sys.stdout = sys.__stdout__ self.assertTrue('abc\t2\nxyz\t1\nabd\t1\nad\t2\nefr\t1' in self.captured_output.getvalue())
class TestFlightsByCarriersReducer(unittest.TestCase): def setUp(self): self.flights_by_carriers_reducer = FlightsByCarriersReducer() self.captured_output = StringIO() sys.stdout = self.captured_output def tearDown(self) -> None: sys.stdout = sys.__stdout__ def test_reduce_good_values(self): sys.stdin = StringIO(INPUT) self.flights_by_carriers_reducer.reduce() sys.stdout = sys.__stdout__ self.assertTrue( 'CO\t3\nPS\t2\nAS\t4' in self.captured_output.getvalue()) def test_reduce_bad_values(self): sys.stdin = StringIO(BAD_INPUT) self.flights_by_carriers_reducer.reduce() sys.stdout = sys.__stdout__ self.assertTrue( 'CO\t2\nPS\t2\nAS\t4' in self.captured_output.getvalue())
class TestWordCountReducer(unittest.TestCase): def setUp(self): self.reducer = WordCountReducer() self.captured_output = StringIO() sys.stdout = self.captured_output def tearDown(self) -> None: sys.stdout = sys.__stdout__ def test_reduce(self): sys.stdin = StringIO( "abc\t1\nabc\t1\nxyz\t1\nabd\t1\nad\t1\nad\t1\nefr\t1") self.reducer.reduce() sys.stdout = sys.__stdout__ self.assertTrue('abc\t2\nxyz\t1\nabd\t1\nad\t2\nefr\t1' in self.captured_output.getvalue())
class TestWordCountMapper(unittest.TestCase): def setUp(self): self.mapper = WordCountMapper() self.captured_output = StringIO() sys.stdout = self.captured_output def tearDown(self): sys.stdout = sys.__stdout__ def test_process_line(self): words = self.mapper.proc_line("abcdef.; -abc xyz-abd ad? efr;") self.assertEqual(words, ['abcdef', 'abc', 'xyz', 'abd', 'ad', 'efr']) def test_map(self): sys.stdin = StringIO("abcdef.; -abc xyz-abd ad? efr;") self.mapper.map() sys.stdout = sys.__stdout__ self.assertTrue('abcdef\t1\nabc\t1\nxyz\t1\nabd\t1\nad\t1\nefr\t1' in self.captured_output.getvalue())
def setUp(self): self.mapper = WordCountMapper() self.captured_output = StringIO() sys.stdout = self.captured_output
def test_map(self): sys.stdin = StringIO("abcdef.; -abc xyz-abd ad? efr;") self.mapper.map() sys.stdout = sys.__stdout__ self.assertTrue('abcdef\t1\nabc\t1\nxyz\t1\nabd\t1\nad\t1\nefr\t1' in self.captured_output.getvalue())
def test_reduce_bad_values(self): sys.stdin = StringIO(BAD_INPUT) self.flights_by_carriers_reducer.reduce() sys.stdout = sys.__stdout__ self.assertTrue( 'CO\t2\nPS\t2\nAS\t4' in self.captured_output.getvalue())
def setUp(self): self.flights_by_carriers_reducer = FlightsByCarriersReducer() self.captured_output = StringIO() sys.stdout = self.captured_output
def setUp(self): self.reducer = WordCountReducer() self.captured_output = StringIO() sys.stdout = self.captured_output