def test_box3d(spark, tmpdir): df = spark.createDataFrame([Row(Box3d(Point(1, 2, 3), 1, 2, 3, 2.5))]) _check_roundtrip(spark, df, tmpdir)
def test_box3d(self): df = self.spark.createDataFrame([Row(Box3d(Point(1, 2, 3), 1, 2, 3, 2.5))]) self._check_roundtrip(df)
def test_point(spark, tmpdir): df = spark.createDataFrame([Row(Point(1, 2, 3)), Row(Point(2, 3, 4))]) _check_roundtrip(spark, df, tmpdir)
def test_point(self): df = self.spark.createDataFrame([Row(Point(1, 2, 3)), Row(Point(2, 3, 4))]) self._check_roundtrip(df)