tests/basics/special_methods: Add testcases for __int__.

xobs · Dec 7, 2018 · d690c2e · d690c2e
1 parent b1d0872
commit d690c2e
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/tests/basics/special_methods.py b/tests/basics/special_methods.py
@@ -93,6 +93,9 @@ def __isub__(self, other):
         print("__isub__ called")
         return self
 
+    def __int__(self):
+        return 42
+
 cud1 = Cud()
 cud2 = Cud()
 
@@ -104,5 +107,16 @@ def __isub__(self, other):
 cud1 > cud2
 cud1 + cud2
 cud1 - cud2
+print(int(cud1))
+
+class BadInt:
+    def __int__(self):
+        print("__int__ called")
+        return None
+
+try:
+    int(BadInt())
+except TypeError:
+    print("TypeError")
 
 # more in special_methods2.py