class TestZn(unittest.TestCase):
def setUp(self):
self.z2 = Zn(2)
self.z3 = Zn(3)
self.z10 = Zn(10)
self.z11 = Zn(11)
self.z100 = Zn(100)
self.z101 = Zn(101)
self.t2 = Zn.TableFormat(2, self.z2.add, range(2), "+")
self.t3 = Zn.TableFormat(3, self.z3.add, range(3), "+")
self.t10 = Zn.TableFormat(10, self.z10.add, range(10), "+")
self.t11 = Zn.TableFormat(11, self.z11.add, range(11), "+")
self.t100 = Zn.TableFormat(100, self.z100.add, range(100), "+")
self.t101 = Zn.TableFormat(101, self.z101.add, range(101), "+")
def test_z2_str(self):
result = self.z2.__str__()
self.assertEqual(result, """Z2
+| 0 1 *| 1
--+---- --+--
0| 0 1 1| 1
1| 1 0 """)
def test_init_error(self):
Zn(2)
self.assertRaises(ValueError, Zn, 1)
def test_col_width(self):
self.assertEqual(self.t2.col_width(), 2)
self.assertEqual(self.t10.col_width(), 2)
self.assertEqual(self.t11.col_width(), 3)
self.assertEqual(self.t100.col_width(), 3)
self.assertEqual(self.t101.col_width(), 4)
def test_cap_data(self):
self.assertEqual(Zn.Cap(self.t2).data(), " 0/ 1".split("/"))
self.assertEqual(Zn.Cap(self.t3).data(), " 0/ 1/ 2".split("/"))
def test_data_data(self):
self.assertEqual(Zn.Data(1, self.t2).data(), " 1/ 0".split("/"))
self.assertEqual(Zn.Data(2, self.t3).data(), " 2/ 0/ 1".split("/"))
def test_cap(self):
self.assertEqual(Zn.StringTable(self.t2).cap(), " +| 0 1")
self.assertEqual(Zn.StringTable(self.t10).cap(), " +| 0 1 2 3 4 5 6 7 8 9")
self.assertEqual(Zn.StringTable(self.t11).cap(), " +| 0 1 2 3 4 5 6 7 8 9 10")
def test_data(self):
self.assertEqual(Zn.Data(1, self.t2).output(), " 1/|/ 1/ 0".split("/"))
self.assertEqual(Zn.Data(2, self.t3).output(), " 2/|/ 2/ 0/ 1".split("/"))
def test_sep(self):
self.assertEqual(Zn.StringTable(self.t2).sep(), "--+----")
self.assertEqual(Zn.StringTable(self.t3).sep(), "--+------")
def test_str_multi2(self):
self.assertEqual(self.z2.str_multi_table(), \
""" *| 1
--+--
1| 1""".split("\n"))
def test_str_multi3(self):
self.assertEqual(self.z3.str_multi_table(), \
""" *| 1 2
--+----
1| 1 2
2| 2 1""".split("\n"))
def test_str_add2(self):
self.assertEqual(self.z2.str_add_table(), \
""" +| 0 1
--+----
0| 0 1
1| 1 0""".split("\n"))
def test_str_add3(self):
self.assertEqual(self.z3.str_add_table(), \
""" +| 0 1 2
--+------
0| 0 1 2
1| 1 2 0
2| 2 0 1""".split("\n"))
def test_mul(self):
self.assertEqual(self.z3.mul(2, 1), 2)
self.assertEqual(self.z3.mul(2, 2), 1)
self.assertEqual(self.z3.mul(2, 3), 0)
def test_add(self):
self.assertEqual(self.z3.add(2, 1), 0)
self.assertEqual(self.z3.add(2, 2), 1)
def test_tableformat_op(self):
self.assertEqual(self.t3.op(2, 1), 0)
self.assertEqual(self.t3.op(2, 2), 1)
