From 3085abd4dc7f57eaf6a464139b89ede802560f5f Mon Sep 17 00:00:00 2001 From: Fernando Erd <fcerd15@inf.ufpr.br> Date: Mon, 8 Jun 2020 11:05:08 -0300 Subject: [PATCH] Fix comma error and generic max year --- regression_test.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/regression_test.py b/regression_test.py index 65a78b0..b2f9e78 100644 --- a/regression_test.py +++ b/regression_test.py @@ -25,6 +25,8 @@ import numpy as np import settings from termcolor import colored,cprint +maxYear = 2019 + class RegressionTest: def __init__(self, route): @@ -41,7 +43,8 @@ class RegressionTest: if len(route) > 1: arguments = arguments.join(route[1:]) file_name = name+arguments - url = self.base_url+name+'?filter=min_year:1991,max_year:2018,'+arguments+'&format=csv' + url = self.base_url+name+'?filter=min_year:1991,max_year:' + str(maxYear) + ','+arguments+'&format=csv' + url = url.replace(",&format","&format") try: api_csv = pd.read_csv(url,float_precision='round_trip', encoding="utf-8-sig") #get from api api_csv.to_csv('route_result/'+file_name+'.csv', encoding="utf-8-sig") #save @@ -57,7 +60,7 @@ class RegressionTest: if len(route) > 1: arguments = arguments.join(route[1:]) file_name = name+arguments - url = self.base_url+name+'?filter=min_year:1991,max_year:2018,'+arguments+'&format=csv' + url = self.base_url+name+'?filter=min_year:1991,max_year:'+str(maxYear)+','+arguments+'&format=csv' try: api_csv = pd.read_csv(url,float_precision='round_trip', encoding="utf-8-sig") #get from api csv_route = pd.read_csv('route_result/'+file_name+'.csv', index_col=0, float_precision='round_trip', encoding="utf-8-sig") #get file -- GitLab