python_unitest_example/test_complex_arithmetic.py at master · innuy/python_unitest_example · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
fromunittestimportTestCase, skip, skipIf

fromcompleximportComplex


classTestComplexArithmetic(TestCase):

defsetUp(self):
self.zero=Complex()

@skip("Sure that this is correct, and will never change")
deftest_default_complex_value(self):
self.assertEqual(self.zero.r, 0)
self.assertEqual(self.zero.i, 0)

deftest_add_complex(self):
one_onei=Complex(1, 1)
result=one_onei+self.zero
self.assertEqual(result.r, 1)
self.assertEqual(result.i, 1)

deftest_sub_complex(self):
one_onei=Complex(1, 1)
result=self.zero-one_onei
self.assertEqual(result.r, -1)
self.assertEqual(result.i, -1)

@skipIf(Complex.__version__< (1, 1), "Not supported by this library")
deftest_equal_complex(self):
zero=Complex()
one=Complex(1)
self.assertTrue(zero==self.zero)
self.assertFalse(self.zero==one)

deftearDown(self):
delself.zero