Add unit test for enum encoder

filmor · filmor · commit 9e70b8f1ef5b · 2022-05-04T08:33:32.000+02:00
diff --git a/tests/test_codec.py b/tests/test_codec.py
@@ -1,39 +1,53 @@
 # -*- coding: utf-8 -*-
 
 """Test conversions using codecs from client python code"""
-import clr
-import System
+
 import pytest
 import Python.Runtime
+import Python.Test as Test
 from Python.Test import ListConversionTester, ListMember, CodecResetter
 
-class int_iterable():
+
+@pytest.fixture(autouse=True)
+def reset():
+    yield
+    CodecResetter.Reset()
+
+
+class int_iterable:
     def __init__(self):
         self.counter = 0
+
     def __iter__(self):
         return self
+
     def __next__(self):
         if self.counter == 3:
             raise StopIteration
         self.counter = self.counter + 1
         return self.counter
 
-class obj_iterable():
+
+class obj_iterable:
     def __init__(self):
         self.counter = 0
+
     def __iter__(self):
         return self
+
     def __next__(self):
         if self.counter == 3:
             raise StopIteration
         self.counter = self.counter + 1
         return ListMember(self.counter, "Number " + str(self.counter))
 
+
 def test_iterable():
-    """Test that a python iterable can be passed into a function that takes an IEnumerable<object>"""
+    """Test that a python iterable can be passed into a function that takes an
+    IEnumerable<object>"""
 
-    #Python.Runtime.Codecs.ListDecoder.Register()
-    #Python.Runtime.Codecs.SequenceDecoder.Register()
+    # Python.Runtime.Codecs.ListDecoder.Register()
+    # Python.Runtime.Codecs.SequenceDecoder.Register()
     Python.Runtime.Codecs.IterableDecoder.Register()
     ob = ListConversionTester()
 
@@ -43,28 +57,33 @@ def test_iterable():
     iterable2 = obj_iterable()
     assert 3 == ob.GetLength2(iterable2)
 
-    CodecResetter.Reset()
 
 def test_sequence():
     Python.Runtime.Codecs.SequenceDecoder.Register()
     ob = ListConversionTester()
 
-    tup = (1,2,3)
+    tup = (1, 2, 3)
     assert 3 == ob.GetLength(tup)
 
     tup2 = (ListMember(1, "one"), ListMember(2, "two"), ListMember(3, "three"))
     assert 3 == ob.GetLength(tup2)
 
-    CodecResetter.Reset()
 
 def test_list():
     Python.Runtime.Codecs.SequenceDecoder.Register()
     ob = ListConversionTester()
 
-    l = [1,2,3]
+    l = [1, 2, 3]
     assert 3 == ob.GetLength(l)
 
     l2 = [ListMember(1, "one"), ListMember(2, "two"), ListMember(3, "three")]
     assert 3 == ob.GetLength(l2)
 
-    CodecResetter.Reset()
+
+def test_enum():
+    Python.Runtime.PyObjectConversions.RegisterEncoder(
+        Python.Runtime.Codecs.EnumPyIntCodec.Instance
+    )
+
+    assert Test.LongEnum.Max == 9223372036854775807
+    assert Test.LongEnum.Min == -9223372036854775808
