Use numbers.Integral on python2

kanghyojun · kanghyojun · commit 94bc1fd8b09d · 2016-11-23T21:50:48.000+09:00
diff --git a/nirum/deserialize.py b/nirum/deserialize.py
@@ -5,6 +5,7 @@
 import datetime
 import decimal
 import enum
+import numbers
 import typing
 import uuid
 
@@ -26,7 +27,7 @@
 )
 _NIRUM_PRIMITIVE_TYPE = {
     float, decimal.Decimal, uuid.UUID, datetime.datetime,
-    datetime.date, bool, int, text_type
+    datetime.date, bool, int, text_type, numbers.Integral
 }
 
 
@@ -122,6 +123,8 @@ def deserialize_primitive(cls, data):
             raise ValueError("'{}' is not a date.".format(data))
     elif cls in {int, float, uuid.UUID, bool}:
         d = cls(data)
+    elif cls is numbers.Integral:
+        d = data
     elif cls is decimal.Decimal:
         try:
             d = cls(data)
diff --git a/tests/deserialize_test.py b/tests/deserialize_test.py
@@ -1,10 +1,11 @@
 import datetime
 import decimal
+import numbers
 import uuid
 import typing
 
 from pytest import raises, mark
-from six import text_type
+from six import PY3, text_type
 
 from nirum._compat import utc
 from nirum.serialize import serialize_record_type
@@ -121,7 +122,7 @@ def test_deserialize_multiple_boxed_type(fx_layered_boxed_types):
 @mark.parametrize(
     'data, t, expect',
     [
-        (1, int, 1),
+        (1, int if PY3 else numbers.Integral, 1),
         (1.1, float, 1.1),
         (u'hello', text_type, 'hello'),
         (True, bool, True),
