More fixes

zjplab · May 26, 2023 · 77fde3a · 77fde3a
1 parent 0d477a2
commit 77fde3a
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/src/google_flight_analysis/scrape.py b/src/google_flight_analysis/scrape.py
@@ -83,7 +83,7 @@ def __repr__(self):
 		else:
 			rep += "{n} RESULTS FOR:\n".format(n = self._data.shape[0])
 
-		for date in self._dates:
+		for date in self._date:
 			rep += "{d}: {org} --> {dest}".format(
 				d = date,
 				org = self._origin,
@@ -219,7 +219,7 @@ def dest(self, x : str) -> None:
 
 	@property
 	def date(self):
-		return self.date
+		return self._date
 
 	@date.setter
 	def date(self, x : str) -> None:
@@ -250,7 +250,7 @@ def _scrape_data(self, driver):
 
 		results = [self._get_results(url, self._data[i], driver) for i, url in enumerate(self._url)]
 
-		self._data = pd.concat(result)
+		self._data = pd.concat(result, ignore_index = True)
 
 		'''if self._date_return is not None:
 			leave_result = self._get_results(self._url[0], driver)

diff --git a/tests/test_class.py b/tests/test_class.py
@@ -13,11 +13,11 @@
 def func_0():
 	return True
 
-res1 = pd.read_csv('tests/test_data/test1.csv')
 res1 = Scrape("LGA", "RDU", "2023-05-15", "2023-06-15")
+res1.data = pd.read_csv('tests/test_data/test1.csv')
 
-res2 = pd.read_csv('tests/test_data/test2.csv')
 res2 = Scrape("IST", "CDG", "2023-07-15", "2023-07-20",)
+res2.data = pd.read_csv('tests/test_data/test2.csv')
 
 os.system('rm tests/test_data/LGA-RDU.csv')
 os.system('rm tests/test_data/CDG-IST.csv')