Fix handling of set()

sebastinas · sebastinas · commit 5b2634e3375c · 2020-10-11T17:03:27.000+02:00
diff --git a/bpython/simpleeval.py b/bpython/simpleeval.py
@@ -111,8 +111,8 @@ def _convert(node):
         elif isinstance(node, ast.Set):
             return set(map(_convert, node.elts))
         elif (
-            isinstance(node, Call)
-            and isinstance(node.func, Name)
+            isinstance(node, ast.Call)
+            and isinstance(node.func, ast.Name)
             and node.func.id == "set"
             and node.args == node.keywords == []
         ):
diff --git a/bpython/test/test_simpleeval.py b/bpython/test/test_simpleeval.py
@@ -2,6 +2,7 @@
 
 import ast
 import numbers
+import sys
 
 from bpython.simpleeval import (
     simple_eval,
@@ -25,6 +26,10 @@ def test_matches_stdlib(self):
     def test_matches_stdlib_py3(self):
         """Should match the stdlib literal_eval if no names or indexing"""
         self.assertMatchesStdlib("{1, 2}")
+
+    @unittest.skipUnless(sys.version_info[:2] >= (3, 9), "Only Python3.9 evaluates set()")
+    def test_matches_stdlib_set_literal(self):
+        """set() is evaluated"""
         self.assertMatchesStdlib("set()")
 
     def test_indexing(self):
