def test_signal_load(self): a = Adapter() a.functionParameters = Mock(return_value=[1, 2, 3]) s = Signal(a, None) s.load() a.functionParameters.assert_called_once_with(None) self.assertEqual(len(s.elements), 3)
def test_method_load(self): a = Adapter() a.functionParameters = Mock(side_effect=[[1, 2, 3], [4, 5]]) m = Method(a, None, None) m.load() a.functionParameters.assert_called_with(None) self.assertEqual(a.functionParameters.call_count, 2) self.assertEqual(len(m.elements), 5)