From b41df1d896fdcb191959fc3f18105f52571a688c Mon Sep 17 00:00:00 2001 From: Daniel Friesel Date: Tue, 17 Sep 2019 07:57:14 +0200 Subject: Mod: Allow initialization from another Mod object --- lib/modular_arithmetic.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/lib/modular_arithmetic.py b/lib/modular_arithmetic.py index 03cb7f8..baf979a 100644 --- a/lib/modular_arithmetic.py +++ b/lib/modular_arithmetic.py @@ -8,6 +8,8 @@ class Mod: __slots__ = ['val','mod'] def __init__(self, val, mod): + if isinstance(val, Mod): + val = val.val if not isinstance(val, int): raise ValueError('Value must be integer') if not isinstance(mod, int) or mod<=0: -- cgit v1.2.3