From 8574291a0b84574626ca213bc6f95dc0db73b0ef Mon Sep 17 00:00:00 2001 From: Bryan Cutler Date: Wed, 29 Aug 2018 13:42:13 -0700 Subject: [PATCH] add test that will output a skipped message when pyarrow is installed --- python/pyspark/sql/tests/test_arrow.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) diff --git a/python/pyspark/sql/tests/test_arrow.py b/python/pyspark/sql/tests/test_arrow.py index 6e75e82d5800..ee054a9b9ac1 100644 --- a/python/pyspark/sql/tests/test_arrow.py +++ b/python/pyspark/sql/tests/test_arrow.py @@ -30,6 +30,21 @@ from pyspark.util import _exception_message +class HaveArrowTests(unittest.TestCase): + + @unittest.skipIf(have_pandas and have_pyarrow, + "Required PyArrow and Pandas were found, Arrow tests will run") + def test_required_pyarrow_pandas_not_installed(self): + # This is only to provide a output when skipped to show that the Arrow tests will run + pass + + @unittest.skipIf(not have_pandas or not have_pyarrow, + "Required PyArrow and Pandas not found, Arrow tests will not run") + def test_required_pyarrow_pandas_installed(self): + # This is only to provide a output when skipped to show that the Arrow tests will not run + pass + + @unittest.skipIf( not have_pandas or not have_pyarrow, pandas_requirement_message or pyarrow_requirement_message)