예제 #1
0
파일: launch.py 프로젝트: hkrd/weather-etl
def get_hottest_day_formatted():
    """
    Finds the hottest day from the two .csv data files, using the output from the Transformer.
    Prints a formatted result which is more convenient for the user.
    """

    transformer = Transformer()
    hottest_day = transformer.find_hottest_day()
    # rename the columns to more understandable names
    hottest_day.columns = ['Date', 'Temperature', 'Region']
    print(
        hottest_day.to_string(formatters={
            "Date":
            lambda x: "{:%d-%m-%Y}".format(pd.to_datetime(x))
        },
                              index=False))
예제 #2
0
class TestTransformer(TestCase):

    def setUp(self):
        self.transformer = Transformer()
        self.project_dir = os.path.abspath(__file__ + "/../../")

    def tearDown(self):
        for f in glob.glob(self.project_dir + "/resources/*.parquet.gzip"):
            os.remove(f)

    def test___init__(self):
        result = self.transformer.find_hottest_day()
        print(result)
        self.assertEqual(1, len(result.index))
        self.assertEqual('2016-03-17T00:00:00', str(result['ObservationDate'].iloc[0]))
        self.assertEqual(15.8, float(result['ScreenTemperature'].iloc[0]))
        self.assertEqual('Highland & Eilean Siar', str(result['Region'].iloc[0]))