From 7708684a55b870fb1d971db94b8dfcce7b2bdba8 Mon Sep 17 00:00:00 2001 From: Wes Roach Date: Sun, 31 Jul 2022 21:50:23 -0500 Subject: [PATCH] Reformat expected output strings to match latest version. --- dendri/tests/test_dataframe.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/dendri/tests/test_dataframe.py b/dendri/tests/test_dataframe.py index 2b4f3ef..079b280 100644 --- a/dendri/tests/test_dataframe.py +++ b/dendri/tests/test_dataframe.py @@ -254,7 +254,7 @@ def test_it_raises_when_struct_field_is_missing1(self, spark_context): validate_schema(source_df, required_schema) assert excinfo.value.args[0] == ( "The following StructFields are not included in the DataFrame:" - " [StructField(city,StringType,true)]" + " [StructField('city', StringType(), True)]" ) def test_it_does_nothing_when_the_schema_matches(self, spark_context): @@ -672,11 +672,11 @@ def test_data_template_from_schema(spark_context): "ArrayType(String), BinaryType, BooleanType, ByteType, DateType, DecimalType, " "DoubleType, FloatType, FractionalType, IntegerType, IntegralType, LongType, " "MapType, NullType, NumericType, StringType, StructType, TimestampType, " - "UserDefinedType, ShortType\n" - '(["String?"], b"", True False, ByteType, date.fromisoformat(""), ' - "DecimalType(10,0), DoubleType, FloatType, FractionalType, IntegerType, " - 'IntegralType, LongType, {"": ""}, None, NumericType, "String", {"": ""}, ' - 'datetime.fromisoformat(""), UDT(), ShortType)' + 'UserDefinedType, ShortType\n(["String?"], b"", True False, ByteType(), ' + 'date.fromisoformat(""), DecimalType(10,0), DoubleType(), FloatType(), ' + 'FractionalType(), IntegerType(), IntegralType(), LongType(), {"": ""}, None, ' + 'NumericType(), "String", {"": ""}, datetime.fromisoformat(""), UDT(), ' + "ShortType())" ) res = data_template_from_schema(schema) assert res == exp