xinrong-meng commented on code in PR #37888: URL: https://github.com/apache/spark/pull/37888#discussion_r971408190
########## python/pyspark/pandas/tests/test_spark_functions.py: ########## @@ -20,25 +20,12 @@ from pyspark.pandas.spark import functions as SF from pyspark.pandas.utils import spark_column_equals from pyspark.sql import functions as F -from pyspark.sql.types import ( - ByteType, - FloatType, - IntegerType, - LongType, -) from pyspark.testing.pandasutils import PandasOnSparkTestCase class SparkFunctionsTests(PandasOnSparkTestCase): - def test_lit(self): - self.assertTrue(spark_column_equals(SF.lit(np.int64(1)), F.lit(1).astype(LongType()))) - self.assertTrue(spark_column_equals(SF.lit(np.int32(1)), F.lit(1).astype(IntegerType()))) - self.assertTrue(spark_column_equals(SF.lit(np.int8(1)), F.lit(1).astype(ByteType()))) - self.assertTrue(spark_column_equals(SF.lit(np.byte(1)), F.lit(1).astype(ByteType()))) - self.assertTrue( - spark_column_equals(SF.lit(np.float32(1)), F.lit(float(1)).astype(FloatType())) - ) - self.assertTrue(spark_column_equals(SF.lit(1), F.lit(1))) + def test_repeat(self): Review Comment: This change is irrelevant to the PR proposal but is made to keep the `test_spark_functions` file. We shall complete tests for `pyspark/pandas/spark/functions` here later. -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: [email protected] For queries about this service, please contact Infrastructure at: [email protected] --------------------------------------------------------------------- To unsubscribe, e-mail: [email protected] For additional commands, e-mail: [email protected]
