Add tests for augmented floor division

diff --git a/Lib/test/output/test_augassign b/Lib/test/output/test_augassign
index ceffd59610eee1ce1f7a1ee7834ae7810ff79540..2a461f63f42b31a3e80b610c2eb7fd93d174d6d3 100644 (file)
--- a/Lib/test/output/test_augassign
+++ b/Lib/test/output/test_augassign
@@ -25,6 +25,9 @@ __imul__ called
 __div__ called
 __rdiv__ called
 __idiv__ called
+__floordiv__ called
+__rfloordiv__ called
+__ifloordiv__ called
 __mod__ called
 __rmod__ called
 __imod__ called

diff --git a/Lib/test/test_augassign.py b/Lib/test/test_augassign.py
index a37b4e1604c5dfe993fc1ce7cb08e723f407b652..e9e5b238649b5a5a40c12f77eb26f6681e090e6e 100644 (file)
--- a/Lib/test/test_augassign.py
+++ b/Lib/test/test_augassign.py
@@ -6,6 +6,7 @@ x *= 2
 x **= 2
 x -= 8
 x /= 2
+x //= 1
 x %= 12
 x &= 2
 x |= 5
@@ -19,6 +20,7 @@ x[0] *= 2
 x[0] **= 2
 x[0] -= 8
 x[0] /= 2
+x[0] //= 2
 x[0] %= 12
 x[0] &= 2
 x[0] |= 5
@@ -32,6 +34,7 @@ x[0] *= 2
 x[0] **= 2
 x[0] -= 8
 x[0] /= 2
+x[0] //= 1
 x[0] %= 12
 x[0] &= 2
 x[0] |= 5
@@ -128,6 +131,23 @@ class testall:
         print "__idiv__ called"
         return self
 
+    def __floordiv__(self, val):
+        print "__floordiv__ called"
+        return self
+    def __ifloordiv__(self, val):
+        print "__ifloordiv__ called"
+        return self
+    def __rfloordiv__(self, val):
+        print "__rfloordiv__ called"
+        return self
+
+    def __truediv__(self, val):
+        print "__truediv__ called"
+        return self
+    def __itruediv__(self, val):
+        print "__itruediv__ called"
+        return self
+
     def __mod__(self, val):
         print "__mod__ called"
     def __rmod__(self, val):
@@ -201,6 +221,10 @@ x / 1
 1 / x
 x /= 1
 
+x // 1
+1 // x
+x //= 1
+
 x % 1
 1 % x
 x %= 1