Merge pull request beeware#682 from onyb/add-sum-builtin-tests

Add more tests for the sum builtin

tests/builtins/test_sum.py

@@ -5,6 +5,7 @@ class SumTests(TranspileTestCase):
     def test_sum_list(self):
         self.assertCodeExecution("""
             print(sum([1, 2, 3, 4, 5, 6, 7]))
+            print(sum([[1, 2], [3, 4], [5, 6]], []))
         """)
 
     def test_sum_tuple(self):
@@ -24,10 +25,29 @@ def test_sum_mix_floats_and_ints(self):
             print(sum([1, 1.414, 2, 3.14159]))
         """)
 
+    def test_sum_frozenset(self):
+        self.assertCodeExecution("""
+            print(sum(frozenset([1, 1.414, 2, 3.14159])))
+        """)
+
+    def test_sum_set(self):
+        self.assertCodeExecution("""
+            print(sum({1, 1.414, 2, 3.14159}))
+        """)
+
+    def test_sum_dict(self):
+        self.assertCodeExecution("""
+            print(sum({1: 1.414, 2: 3.14159}))
+        """)
+
+    def test_sum_generator_expressions(self):
+        self.assertCodeExecution("""
+            print(sum(x ** 2 for x in [3, 4]))
+        """)
+
 
 class BuiltinSumFunctionTests(BuiltinFunctionTestCase, TranspileTestCase):
     functions = ["sum"]
 
     not_implemented = [
-        'test_frozenzet',
     ]