Allow route_stops to be an Iterable instead of a Sequence

Author: ViViDboarder

py_nextbus/client.py

@@ -4,7 +4,6 @@
 import logging
 import urllib.request
 from collections.abc import Iterable
-from collections.abc import Sequence
 from datetime import datetime
 from enum import Enum
 from typing import Any
@@ -105,7 +104,7 @@ def get_messages(
         messages to display for multiple routes.
 
         Arguments:
-            route_tags: (Sequence of Strings) List of NextBus route tags for the routes to get
+            route_tags: (Iterable of Strings) List of NextBus route tags for the routes to get
                 messages for.
             agency: (String) Name of a transit agency on NextBus. This must be provided if the
                 "agency" instance attribute has not been set.
@@ -122,7 +121,7 @@ def get_messages(
         """
 
         if not isinstance(route_tags, Iterable) or isinstance(route_tags, str):
-            raise TypeError('"route_tags" must be a Sequence but not a single string.')
+            raise TypeError('"route_tags" must be a Iterable but not a single string.')
 
         agency = self._get_agency(agency)
 
@@ -234,7 +233,7 @@ def get_predictions_for_multi_stops(
         arrival time predictions for multiple stops.
 
         Arguments:
-            route_stops: (Sequence of RouteStops) Sequence of tuples identifying the combinations of
+            route_stops: (Iterable of RouteStops) Iterable of tuples identifying the combinations of
                 routes and stops to get predictions for.
             agency: (String) Name of a transit agency on NextBus. This must be provided if the
                 "agency" instance attribute has not been set.
@@ -250,7 +249,7 @@ def get_predictions_for_multi_stops(
                 valid JSON.
         """
 
-        if not isinstance(route_stops, Sequence) or isinstance(route_stops, str):
+        if not isinstance(route_stops, Iterable) or isinstance(route_stops, str):
             raise TypeError('"route_stops" must be a sequence.')
 
         agency = self._get_agency(agency)

tests/client_test.py

@@ -124,6 +124,7 @@ def test_type_error_not_raised_if_route_tags_is_iterable(
 
         nextbus_client = client.NextBusClient(output_format=client.ReturnFormat.JSON)
         nextbus_client.get_messages(route_tags=["foo"], agency="bar")
+        nextbus_client.get_messages(route_tags={"foo"}, agency="bar")
         nextbus_client.get_messages(route_tags=("foo",), agency="bar")
         nextbus_client.get_messages(route_tags=(f for f in ("foo",)), agency="bar")
 
@@ -256,12 +257,6 @@ def test_type_error_raised_if_stops_is_not_iterable(
 
         nextbus_client = client.NextBusClient(output_format=client.ReturnFormat.JSON)
 
-        with self.assertRaises(TypeError):
-            nextbus_client.get_predictions_for_multi_stops(route_stops={})
-
-        with self.assertRaises(TypeError):
-            nextbus_client.get_predictions_for_multi_stops(route_stops=set())
-
         with self.assertRaises(TypeError):
             nextbus_client.get_predictions_for_multi_stops(route_stops=None)