From ac6f79c599b4d5f8cad4651395712b1920aafead Mon Sep 17 00:00:00 2001
From: sanjeevtw <30316618+sanjeevtw@users.noreply.github.com>
Date: Tue, 19 Apr 2022 16:17:22 +0530
Subject: [PATCH] Update Data Transformation CO2 vs Temperature.py

Fix the test method for fix country
---
 .../Data Transformation CO2 vs Temperature.py             | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/data-transformation/Data Transformation CO2 vs Temperature.py b/data-transformation/Data Transformation CO2 vs Temperature.py
index 1b8b221..89bf4d9 100644
--- a/data-transformation/Data Transformation CO2 vs Temperature.py	
+++ b/data-transformation/Data Transformation CO2 vs Temperature.py	
@@ -587,10 +587,10 @@ def fix_country(col: Column) -> Column:
 
 # COMMAND ----------
 
-def test_fix_country(self):
+def test_fix_country():
   original = pd.Series(["  gErMaNy ", "   uNiTeD sTaTeS    "])
-  spark_df = self.spark.createDataFrame(pd.DataFrame({"Country": original}))
-  spark_df = spark_df.withColumn("Country", self.transformer.fix_country(F.col("Country")))
+  spark_df = spark.createDataFrame(pd.DataFrame({"Country": original}))
+  spark_df = spark_df.withColumn("Country", fix_country(F.col("Country")))
   fixed = spark_df.toPandas()
   try:
     result = sorted(fixed["Country"])
@@ -598,6 +598,8 @@ def test_fix_country(self):
   except Exception as e:
     raise type(e)(''.join(debug(original))) from e
   print("All tests pass! :)")
+  
+test_fix_country()
 
 # COMMAND ----------