def test_stops_creates_generator(self): """ Remove stops creates a generator """ inputs = ['test', 'the'] expected = ['test'] stops = ['the'] result = remove_stops(inputs, stops) self.assertIsInstance(result, type(x for x in list()))
def test_stops(self): inputs = ['test', 'the', '123'] inputs.extend(string.punctuation) expected = ['test'] stops = ['the'] result = remove_stops(inputs, stops) self.assertEqual(list(result), expected)