Bugfix for get_range_coordinates_air

src/ozone/ozone.py

@@ -228,7 +228,9 @@ def _extract_forecast_data(self, data_obj: Any) -> pandas.DataFrame:
  df.index = pandas.to_datetime(df.index)
  return df
 
- def _check_and_get_data_obj(self, r: requests.Response, **check_debug_info) -> dict:
+ def _check_and_get_data_obj(
+ self, r: requests.Response, **check_debug_info
+ ) -> Union[dict, List[dict]]:
  """Get data object from API response and throw error if any is encouuntered
 
  Args:
@@ -239,8 +241,8 @@ def _check_and_get_data_obj(self, r: requests.Response, **check_debug_info) -> d
  city names to show it instead of just generic exception message.
 
  Returns:
- dict: The data object i.e. the `data` part of the API response,
- in dictionary format (already JSON-ified).
+ Union[dict, List[dict]]: The data object i.e. the `data` part of the
+ API response, in dictionary or list format (already JSON-ified).
 
  """
  self._check_status_code(r)
@@ -249,8 +251,11 @@ def _check_and_get_data_obj(self, r: requests.Response, **check_debug_info) -> d
  status = response.get("status")
  data = response.get("data")
 
- if status == "ok" and isinstance(data, dict):
- return data
+ if status == "ok":
+ if isinstance(data, dict) or isinstance(data, list):
+ # Only return data if status is ok and data is either dict or list.
+ # Otherwise it gets to exception raisers below.
+ return data
 
  if isinstance(data, str):
  if "Unknown station" in data:

tests/cassettes/test_get_range_coordinates_air/test_bad_coordinates.yaml

@@ -24,15 +24,15 @@ interactions:
  Content-Type:
  - application/json; charset=UTF-8
  Date:
- - Fri, 22 Apr 2022 01:27:05 GMT
+ - Fri, 22 Apr 2022 17:39:52 GMT
  Server:
  - nginx
  Transfer-Encoding:
  - chunked
  Vary:
  - Accept-Encoding
  X-Gen-Time:
- - "33.192\xC2\xB5s"
+ - "24.849\xC2\xB5s"
  X-Powered-By:
  - rxstreamer-waqi/1.3
  content-length:

tests/test_get_range_coordinates_air.py

@@ -25,8 +25,8 @@ def test_column_expected_contents():
  # Sanity check: make sure lat-lon are within range
  assert (result["latitude"] > LOWER_BOUND[0]).all()
  assert (result["latitude"] < UPPER_BOUND[0]).all()
- assert (result["longitude"] < LOWER_BOUND[1]).all()
- assert (result["longitude"] > UPPER_BOUND[1]).all()
+ assert (result["longitude"] > LOWER_BOUND[1]).all()
+ assert (result["longitude"] < UPPER_BOUND[1]).all()
 
  # Range of coordinates: no city was given, hence city must all be nan
  assert numpy.isnan(result["city"]).all()