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)