add: __repr__ to ElementInFp

syakoo · Nov 13, 2020 · 315eee0 · 315eee0
1 parent 807563f
commit 315eee0
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/galois_field/ElementInGFp.py b/galois_field/ElementInGFp.py
@@ -36,6 +36,9 @@ def inverse(self) -> ElementInGFp:
     def __str__(self) -> str:
         return f'{str(self.value)} (mod {self.p})'
 
+    def __repr__(self) -> str:
+        return f'ElementInGFp({self.value}, {self.p})'
+
     def __add__(self, other: ElementInGFp) -> ElementInGFp:
         return ElementInGFp(self.value + other.value, self.p)
 

diff --git a/tests/test_ElementInGFp.py b/tests/test_ElementInGFp.py
@@ -26,6 +26,17 @@ def test_GFElementInGFp_str(value, p, expected):
     assert str(result) == expected
 
 
+@pytest.mark.parametrize('value, p, expected', [
+    (1, 5, 'ElementInGFp(1, 5)'),
+    (222, 11, 'ElementInGFp(2, 11)'),
+    (-2, 123456791, 'ElementInGFp(123456789, 123456791)')
+])
+def test_GFElementInGFp_repr(value, p, expected):
+    result = ElementInGFp(value, p)
+
+    assert repr(result) == expected
+
+
 @pytest.mark.parametrize('value1, value2, p, expected_value', [
     (1, 1, 5, 2),
     (111, 111, 11, 2),