type checking fix for Python 2.7?

MinchinWeb · Jul 20, 2015 · f181ebe · f181ebe
1 parent aef5c8a
commit f181ebe
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/colourettu/_colour.py b/colourettu/_colour.py
@@ -239,7 +239,8 @@ def _luminance(mycolour):
         \\[ lum = \\sqrt{0.299 r^2 + 0.587 g^2 + 0.114 b^2} \\]
     '''
 
-    if type(mycolour) is colour:
+    colour_for_type = colour()
+    if type(mycolour) is type(colour_for_type):
         mycolour2 = mycolour
     else:
         try:
@@ -318,15 +319,16 @@ def _contrast(colour1, colour2):
     any quantitative research to this effect.
     '''
 
-    if type(colour1) is colour:
+    colour_for_type = colour()
+    if type(colour1) is type(colour_for_type):
         mycolour1 = colour1
     else:
         try:
             mycolour1 = colour(colour1)
         except:
             raise TypeError("colour1 must be a colourettu.colour")
 
-    if type(colour2) is colour:
+    if type(colour2) is type(colour_for_type):
         mycolour2 = colour2
     else:
         try: