Remove trailing whitespace.

[cython.git] / tests / run / if.pyx

diff --git a/tests/run/if.pyx b/tests/run/if.pyx
index f7c120a6535bcf84e66afba70473b9d243af72ae..f1c3d5ab89eeff808b6b17769ab22e033a575b45 100644 (file)
--- a/tests/run/if.pyx
+++ b/tests/run/if.pyx
@@ -1,48 +1,70 @@
-__doc__ = u"""
+def f(a, b):
+    """
     >>> f(0,0)
     0
     >>> f(1,2)
     2
     >>> f(1,-1)
     1
+    """
+    x = 0
+    if a:
+        x = 1
+    if a+b:
+        x = 2
+    return x
 
+def g(a, b):
+    """
     >>> g(1,2)
     1
     >>> g(0,2)
     2
     >>> g(0,0)
     0
+    """
+    x = 0
+    if a:
+        x = 1
+    elif b:
+        x = 2
+    return x
 
+def h(a, b):
+    """
     >>> h(1,2)
     1
     >>> h(0,2)
     2
     >>> h(0,0)
     3
-"""
-
-def f(a, b):
-    x = 0
-    if a:
-        x = 1
-    if a+b:
-        x = 2
-    return x
-
-def g(a, b):
+    """
     x = 0
     if a:
         x = 1
     elif b:
         x = 2
+    else:
+        x = 3
     return x
-    
-def h(a, b):
+
+try:
+    import __builtin__  as builtins
+except ImportError:
+    import builtins
+
+def i(a, b):
+    """
+    >>> i(1,2)
+    1
+    >>> i(2,2)
+    2
+    >>> i(2,1)
+    0
+    """
     x = 0
-    if a:
+    if builtins.str(a).upper() == u"1":
         x = 1
-    elif b:
+    if builtins.str(a+b).lower() not in (u"1", u"3"):
         x = 2
-    else:
-        x = 3
     return x