def test_str(self): galois.set_modulus(25, [1, 0, 2]) a = Polynomial([1, 0, 23, 00, 12]) self.assertEqual(str(a), '1 0 23 0 12') galois.set_modulus(3, [1, 0, 2])
import unittest import galois from galois import Element, Polynomial, PolynomialOnRing galois.set_modulus(3, [1, 0, 2]) class TestElement(unittest.TestCase): def test_init(self): a = Element(-1) self.assertEqual(a.element, 2) a = Element(4) self.assertEqual(a.element, 1) def test_complement(self): a = Element(1) c = a.complement() self.assertEqual(c.element, 2) def test_add(self): a = Element(1) b = Element(1) c = a+b self.assertEqual(c.element, 2) a = Element(2) b = Element(2) c = a+b self.assertEqual(c.element, 1)