further skip

Author: zhengruifeng

python/pyspark/sql/tests/streaming/test_streaming_foreach.py

@@ -16,7 +16,9 @@
 #
 
 import os
+import platform
 import tempfile
+import unittest
 
 from pyspark.testing.sqlutils import ReusedSQLTestCase
 
@@ -278,12 +280,14 @@ class WriterWithNonCallableClose(WithProcess):
         tester.assert_invalid_writer(WriterWithNonCallableClose(), "ATTRIBUTE_NOT_CALLABLE")
 
 
+@unittest.skipIf(
+    "pypy" in platform.python_implementation().lower(), "cannot run in environment pypy"
+)
 class StreamingTestsForeach(StreamingTestsForeachMixin, ReusedSQLTestCase):
     pass
 
 
 if __name__ == "__main__":
-    import unittest
     from pyspark.sql.tests.streaming.test_streaming_foreach import *  # noqa: F401
 
     try:

python/pyspark/streaming/tests/test_listener.py

@@ -14,10 +14,17 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
+
+import platform
+import unittest
+
 from pyspark.streaming import StreamingListener
 from pyspark.testing.streamingutils import PySparkStreamingTestCase
 
 
+@unittest.skipIf(
+    "pypy" in platform.python_implementation().lower(), "cannot run in environment pypy"
+)
 class StreamingListenerTests(PySparkStreamingTestCase):
     duration = 0.5
 
@@ -147,7 +154,6 @@ def func(dstream):
 
 
 if __name__ == "__main__":
-    import unittest
     from pyspark.streaming.tests.test_listener import *  # noqa: F401
 
     try: