add tests

Author: timfel

graalpython/com.oracle.graal.python.test/src/tests/test_map.py

@@ -58,3 +58,25 @@ def test_map2():
     items0 = [0, 1, 2, 3, 4]
     items1 = [5, 6, 7, 8, 9]
     assert list(map(lambda x, y: x * y, items0, items1)) == [0, 6, 14, 24, 36]
+
+
+def test_map_contains():
+    assert 1 in map(lambda s: s, [1])
+    assert 2 not in map(lambda s: s, [1])
+
+    class X():
+        def __iter__(self):
+            return self
+
+        def __next__(self):
+            i = getattr(self, "i", 0)
+            if i == 1:
+                raise StopIteration
+            else:
+                self.i = i + 1
+                return i
+
+    # the below checks that __contains__ consumes the iterator
+    m = map(lambda s: s, X())
+    assert 0 in m
+    assert 0 not in m

graalpython/com.oracle.graal.python.test/src/tests/test_parser.py

@@ -84,3 +84,7 @@ def test_lambda_no_args_with_nested_lambdas():
     except Exception as e:
         no_err = False
     assert no_err
+
+
+def test_byte_numeric_escapes():
+    assert eval('b"PK\\005\\006\\00\\11\\22\\08"') == b'PK\x05\x06\x00\t\x12\x008'

graalpython/com.oracle.graal.python.test/src/tests/test_string.py

@@ -53,7 +53,7 @@ def test_find():
     assert s.find('cau', MyIndexable(5), None) == 5
     assert s.find('cau', MyIndexable(5), MyIndexable(8)) == 5
     assert s.find('cau', None, MyIndexable(8)) == 5
-    
+
 
 def test_rfind():
     assert "test string test".rfind("test") == 12
@@ -749,7 +749,7 @@ def test_zfill(self):
         self.checkequal('0034', '34', 'zfill', 4)
 
         self.checkraises(TypeError, '123', 'zfill')
-        
+
     def test_zfill_specialization(self):
         self.checkequal('123', '123', 'zfill', True)
         self.checkequal('0123', '123', 'zfill', MyIndexable(4))
@@ -880,7 +880,7 @@ def test_count(self):
                 if rem or r1 != r2:
                     self.assertEqual(rem, 0, '%s != 0 for %s' % (rem, i))
                     self.assertEqual(r1, r2, '%s != %s for %s' % (r1, r2, i))
-                    
+
 
 def test_same_id():
     empty_ids = set([id(str()) for i in range(100)])
@@ -889,3 +889,21 @@ def test_same_id():
     assert len(empty_ids) == 1
     empty_ids = set([id(u'') for i in range(100)])
     assert len(empty_ids) == 1
+
+
+def test_translate():
+    assert "abc".translate({ord("a"): "b"}) == "bbc"
+    assert "abc".translate({ord("a"): "xxx"}) == "xxxbc"
+    assert "abc".translate({ord("a"): ""}) == "bc"
+    try:
+        "abc".translate({ord("a"): 8**63})
+    except ValueError as e:
+        assert "mapping must be in range" in str(e)
+    else:
+        assert False, "should raise"
+
+
+def test_splitlines():
+    assert len(str.splitlines("\n\n")) == 2
+    assert len(str.splitlines("\n")) == 1
+    assert len(str.splitlines("a\nb")) == 2

graalpython/com.oracle.graal.python.test/src/tests/test_vars.py

@@ -74,3 +74,16 @@ def test_vars():
     assert_raises(TypeError, vars, 42, 42)
     assert_raises(TypeError, vars, 42)
     assert vars(C_get_vars()) == {'a': 2}
+
+
+def test_vars_update():
+    class X():
+        pass
+
+    x = X()
+    vars(x).update({"args": 42})
+    assert vars(x)["args"] == 42
+    assert x.__dict__["args"] == 42
+
+    vars().update({"arg": 12})
+    assert locals()["arg"] == 12