def test_lists(self): with self._assert_stdout('[2.4, 6.8]\n'): lists.main([1.2, 3.4], 2) with self._assert_stdout('[2, 4, 6]\n'): lists.main([1, 2, 3], 2)
def test_lists(self, print_): lists.main([1.2, 3.4], 2) print_.assert_called_with([2.4, 6.8]) lists.main([1, 2, 3], 2) print_.assert_called_with([2, 4, 6])