From 988b953e5a2e02943d8575aee367ded5ebb7b40f Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Fri, 9 May 2014 23:21:31 -0300
Subject: [PATCH] Treat 'long' correctly in traits()

---
 pymbolic/traits.py | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

diff --git a/pymbolic/traits.py b/pymbolic/traits.py
index 8ac544f..8cbff25 100644
--- a/pymbolic/traits.py
+++ b/pymbolic/traits.py
@@ -40,11 +40,12 @@ def traits(x):
     try:
         return x.traits()
     except AttributeError:
-        if isinstance(x, (complex, float)): return FieldTraits()
-        if isinstance(x, int): return IntegerTraits()
-        raise NoTraitsError
-
-
+        if isinstance(x, (complex, float)):
+            return FieldTraits()
+        elif isinstance(x, (int, long)):
+            return IntegerTraits()
+        else:
+            raise NoTraitsError
 
 
 def common_traits(*args):
-- 
GitLab