diff --git a/sql/core/src/test/scala/org/apache/spark/sql/SubquerySuite.scala b/sql/core/src/test/scala/org/apache/spark/sql/SubquerySuite.scala
index 4ec85b0ac6d2..38a51474cbcd 100644
--- a/sql/core/src/test/scala/org/apache/spark/sql/SubquerySuite.scala
+++ b/sql/core/src/test/scala/org/apache/spark/sql/SubquerySuite.scala
@@ -1388,6 +1388,8 @@ class SubquerySuite extends QueryTest with SharedSQLContext {
   test("SPARK-28441: COUNT bug in WHERE clause (Filter) with PythonUDF") {
     import IntegratedUDFTestUtils._
 
+    assume(shouldTestPythonUDFs)
+
     val pythonTestUDF = TestPythonUDF(name = "udf")
     registerTestUDF(pythonTestUDF, spark)
 
@@ -1409,6 +1411,8 @@ class SubquerySuite extends QueryTest with SharedSQLContext {
   test("SPARK-28441: COUNT bug in SELECT clause (Project) with PythonUDF") {
     import IntegratedUDFTestUtils._
 
+    assume(shouldTestPythonUDFs)
+
     val pythonTestUDF = TestPythonUDF(name = "udf")
     registerTestUDF(pythonTestUDF, spark)
 
@@ -1421,6 +1425,8 @@ class SubquerySuite extends QueryTest with SharedSQLContext {
   test("SPARK-28441: COUNT bug in HAVING clause (Filter) with PythonUDF") {
     import IntegratedUDFTestUtils._
 
+    assume(shouldTestPythonUDFs)
+
     val pythonTestUDF = TestPythonUDF(name = "udf")
     registerTestUDF(pythonTestUDF, spark)
 
@@ -1440,6 +1446,8 @@ class SubquerySuite extends QueryTest with SharedSQLContext {
   test("SPARK-28441: COUNT bug in Aggregate with PythonUDF") {
     import IntegratedUDFTestUtils._
 
+    assume(shouldTestPythonUDFs)
+
     val pythonTestUDF = TestPythonUDF(name = "udf")
     registerTestUDF(pythonTestUDF, spark)
 
@@ -1459,6 +1467,8 @@ class SubquerySuite extends QueryTest with SharedSQLContext {
   test("SPARK-28441: COUNT bug negative examples with PythonUDF") {
     import IntegratedUDFTestUtils._
 
+    assume(shouldTestPythonUDFs)
+
     val pythonTestUDF = TestPythonUDF(name = "udf")
     registerTestUDF(pythonTestUDF, spark)
 
@@ -1487,6 +1497,8 @@ class SubquerySuite extends QueryTest with SharedSQLContext {
   test("SPARK-28441: COUNT bug in nested subquery with PythonUDF") {
     import IntegratedUDFTestUtils._
 
+    assume(shouldTestPythonUDFs)
+
     val pythonTestUDF = TestPythonUDF(name = "udf")
     registerTestUDF(pythonTestUDF, spark)
 
@@ -1507,6 +1519,8 @@ class SubquerySuite extends QueryTest with SharedSQLContext {
   test("SPARK-28441: COUNT bug with nasty predicate expr with PythonUDF") {
     import IntegratedUDFTestUtils._
 
+    assume(shouldTestPythonUDFs)
+
     val pythonTestUDF = TestPythonUDF(name = "udf")
     registerTestUDF(pythonTestUDF, spark)
 
@@ -1525,6 +1539,8 @@ class SubquerySuite extends QueryTest with SharedSQLContext {
   test("SPARK-28441: COUNT bug with attribute ref in subquery input and output with PythonUDF") {
     import IntegratedUDFTestUtils._
 
+    assume(shouldTestPythonUDFs)
+
     val pythonTestUDF = TestPythonUDF(name = "udf")
     registerTestUDF(pythonTestUDF, spark)