def test_that_empty_list_is_returned_with_zero_arg(self): res = multi_gen(0) self.assertFalse(self.mock_generate.called) self.assertEqual(res, [])
def test_called_once_without_parameter(self): multi_gen() self.assertEqual(self.mock_generate.call_count, 1)
def test_called_ntimes_according_to_arg(self): multi_gen(10) self.assertEqual(self.mock_generate.call_count, 10)
def test_that_returns_correct_result_for_one_call(self): self.mock_generate.return_value = '3.2.1' res = multi_gen() self.assertEqual(res, [self.mock_generate.return_value])