def test_basic(self): demo = Demo(1, 2) self.assertEqual(1, demo.num1) self.assertEqual(2, demo.num2) self.assertEqual(1, demo._special_number()) self.assertEqual(4, demo.sum()) self.assertEqual(8, demo.sum_bias())
def test_raise_error_side_effect(self): demo = Demo(1, 2) demo._special_number = MagicMock(side_effect=SpecialError) with self.assertRaises(SpecialError): demo.sum() demo._special_number.assert_called_with()
def test_raise_error(self): demo = Demo(1, 2) demo._special_number = MagicMock(return_value=-1) with self.assertRaises(ValueError): demo.sum() demo._special_number.assert_called_with()