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