from mypy.test.data import DataDrivenTestCase, DataSuite
from mypy.test.testcheck import TypeCheckSuite


class PySparkCoreSuite(TypeCheckSuite):
    TypeCheckSuite.files = ["context.test", "udf.test"]
    required_out_section = True

@@ -167,16 +167,25 @@
 iles = %5B
+%0A        
 %22context
@@ -191,16 +191,55 @@
 t.test%22,
+%0A        %22resultiterable.test%22,%0A       
  %22udf.te
@@ -241,16 +241,22 @@
 df.test%22
+,%0A    
 %5D%0A    re

Enable resultiterable test