fixed tests und

python/pyspark/serializers.py

@@ -373,6 +373,9 @@ def __next__(self):
         else:
             raise ValueError('Received invalid stream status {0}'.format(stream_status))
 
+    def next(self):
+        return self.__next__()
+
     def _read_df(self):
         import pyarrow as pa
         reader = pa.ipc.open_stream(self._stream)

python/pyspark/sql/cogroup.py

@@ -1,3 +1,20 @@
+#
+# Licensed to the Apache Software Foundation (ASF) under one or more
+# contributor license agreements.  See the NOTICE file distributed with
+# this work for additional information regarding copyright ownership.
+# The ASF licenses this file to You under the Apache License, Version 2.0
+# (the "License"); you may not use this file except in compliance with
+# the License.  You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
 from pyspark.sql.dataframe import DataFrame
 
 

python/pyspark/sql/tests/test_pandas_udf_cogrouped_map.py

@@ -144,3 +144,13 @@ def merge_pandas(l, r):
 
         assert_frame_equal(expected, result, check_column_type=_check_column_type)
 
+
+if __name__ == "__main__":
+    from pyspark.sql.tests.test_pandas_udf_cogrouped_map import *
+
+    try:
+        import xmlrunner
+        testRunner = xmlrunner.XMLTestRunner(output='target/test-reports', verbosity=2)
+    except ImportError:
+        testRunner = None
+    unittest.main(testRunner=testRunner, verbosity=2)