Update Range.__getitem__ for boolean indexes.

julianrolon · Aug 18, 2017 · 8e2f7b1 · 8e2f7b1
1 parent f7c1526
commit 8e2f7b1
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/python/common/org/python/types/Range.java b/python/common/org/python/types/Range.java
@@ -86,7 +86,12 @@ public org.python.Object __getitem__(org.python.Object index) {
                 return new org.python.types.Range(substart, substop, substep);
             } else {
                 long len = ((org.python.types.Int) (this.__len__())).value;
-                long idx = ((org.python.types.Int) index).value;
+                long idx;
+                if (index instanceof org.python.types.Bool) {
+                    idx = ((org.python.types.Bool)index).value ? 1 : 0;
+                } else {
+                    idx = ((org.python.types.Int) index).value;
+                }
 
                 if (idx < 0) {
                     idx = len + idx;

diff --git a/tests/datatypes/test_range.py b/tests/datatypes/test_range.py
@@ -88,10 +88,6 @@ class UnaryRangeOperationTests(UnaryOperationTestCase, TranspileTestCase):
 class BinaryRangeOperationTests(BinaryOperationTestCase, TranspileTestCase):
     data_type = 'range'
 
-    not_implemented = [
-        'test_subscr_bool',
-    ]
-
     not_implemented_versions = {
         'test_subscr_None': (3.4,),
         'test_subscr_NotImplemented': (3.4,),