diff --git a/tests/test_constants.py b/tests/test_constants.py index c667282..ed3951f 100644 --- a/tests/test_constants.py +++ b/tests/test_constants.py @@ -1,4 +1,4 @@ -alonso = { +ALONSO = { "driverId":"alonso", "permanentNumber":"14", "code":"ALO", @@ -9,7 +9,7 @@ alonso = { "nationality":"Spanish" } -istanbul = { +ISTANBUL = { "circuitId":"istanbul", "url":"http://en.wikipedia.org/wiki/Istanbul_Park", "circuitName":"Istanbul Park", @@ -21,16 +21,35 @@ istanbul = { } } -ferrari = { +FERRARI = { "constructorId":"ferrari", "url":"http://en.wikipedia.org/wiki/Scuderia_Ferrari", "name":"Ferrari", "nationality":"Italian" } -renault = { +RENAULT = { "constructorId":"renault", "url":"http://en.wikipedia.org/wiki/Renault_in_Formula_One", "name":"Renault", "nationality":"French" +} + +ALPINE = { + "constructorId":"alpine", + "url":"http://en.wikipedia.org/wiki/Alpine_F1_Team", + "name":"Alpine F1 Team", + "nationality":"French" +} + +SILVERSTONE = { + "circuitId":"silverstone", + "url":"http://en.wikipedia.org/wiki/Silverstone_Circuit", + "circuitName":"Silverstone Circuit", + "Location":{ + "lat":"52.0786", + "long":"-1.01694", + "locality":"Silverstone", + "country":"UK" + } } \ No newline at end of file diff --git a/tests/test_requester.py b/tests/test_requester.py index 197a874..852252b 100644 --- a/tests/test_requester.py +++ b/tests/test_requester.py @@ -5,6 +5,8 @@ import tests.test_constants as test_constants class TestRequester(): """ Tests for the Requester class + ~~~ + """ r = ergast_py.Requester() @@ -44,7 +46,7 @@ class TestRequester(): def test_get_circuit(self): - expected = [test_constants.istanbul] + expected = [test_constants.ISTANBUL] params = self._construct_test_params(season=2008, round=5) @@ -52,7 +54,7 @@ class TestRequester(): def test_get_constructors(self): - expected = [test_constants.ferrari] + expected = [test_constants.FERRARI] params = self._construct_test_params(constructor="ferrari") @@ -60,13 +62,13 @@ class TestRequester(): def test_get_drivers(self): - expected = [test_constants.alonso] + expected = [test_constants.ALONSO] params = self._construct_test_params(driver="alonso") assert self.r.get_drivers(params) == expected - + def test_get_qualifying(self): expected = [ { @@ -74,15 +76,15 @@ class TestRequester(): "round":"5", "url":"http://en.wikipedia.org/wiki/2008_Turkish_Grand_Prix", "raceName":"Turkish Grand Prix", - "Circuit": test_constants.istanbul, + "Circuit": test_constants.ISTANBUL, "date":"2008-05-11", "time":"12:00:00Z", "QualifyingResults":[ { "number":"5", "position":"7", - "Driver": test_constants.alonso, - "Constructor": test_constants.renault, + "Driver": test_constants.ALONSO, + "Constructor": test_constants.RENAULT, "Q1":"1:26.836", "Q2":"1:26.522", "Q3":"1:28.422" @@ -93,4 +95,248 @@ class TestRequester(): params = self._construct_test_params(season=2008, round=5, qualifying=7) - assert self.r.get_qualifying(params) == expected \ No newline at end of file + assert self.r.get_qualifying(params) == expected + + + def test_get_sprints(self): + expected = [ + { + "season":"2021", + "round":"10", + "url":"http://en.wikipedia.org/wiki/2021_British_Grand_Prix", + "raceName":"British Grand Prix", + "Circuit": test_constants.SILVERSTONE, + "date":"2021-07-18", + "time":"14:00:00Z", + "SprintResults":[ + { + "number":"14", + "position":"7", + "positionText":"7", + "points":"0", + "Driver": test_constants.ALONSO, + "Constructor": test_constants.ALPINE, + "grid":"11", + "laps":"17", + "status":"Finished", + "Time":{ + "millis":"1581953", + "time":"+43.527" + }, + "FastestLap":{ + "lap":"17", + "Time":{ + "time":"1:31.773" + } + } + } + ] + } + ] + + params = self._construct_test_params(season=2021, round=10, sprint=7) + + assert self.r.get_sprints(params) == expected + + + def test_get_results(self): + expected = [ + { + "season":"2021", + "round":"16", + "url":"http://en.wikipedia.org/wiki/2021_Turkish_Grand_Prix", + "raceName":"Turkish Grand Prix", + "Circuit": test_constants.ISTANBUL, + "date":"2021-10-10", + "time":"12:00:00Z", + "Results":[ + { + "number":"14", + "position":"16", + "positionText":"16", + "points":"0", + "Driver": test_constants.ALONSO, + "Constructor": test_constants.ALPINE, + "grid":"5", + "laps":"57", + "status":"+1 Lap", + "FastestLap":{ + "rank":"14", + "lap":"55", + "Time":{ + "time":"1:33.252" + }, + "AverageSpeed":{ + "units":"kph", + "speed":"206.073" + } + } + } + ] + } + ] + + params = self._construct_test_params(season=2021, round=16, result=16) + + assert self.r.get_results(params) == expected + + + def test_get_races(self): + expected = [ + { + "season":"2021", + "round":"16", + "url":"http://en.wikipedia.org/wiki/2021_Turkish_Grand_Prix", + "raceName":"Turkish Grand Prix", + "Circuit": test_constants.ISTANBUL, + "date":"2021-10-10", + "time":"12:00:00Z", + "FirstPractice":{ + "date":"2021-10-08" + }, + "SecondPractice":{ + "date":"2021-10-08" + }, + "ThirdPractice":{ + "date":"2021-10-09" + }, + "Qualifying":{ + "date":"2021-10-09" + } + } + ] + + params = self._construct_test_params(season=2021, round=16) + + assert self.r.get_races(params) == expected + + + def test_get_seasons(self): + expected = [ + { + "season":"2021", + "url":"http://en.wikipedia.org/wiki/2021_Formula_One_World_Championship" + } + ] + + params = self._construct_test_params(season=2021) + + assert self.r.get_seasons(params) == expected + + + def test_get_statuses(self): + expected = [ + { + "statusId":"11", + "count":"1", + "status":"+1 Lap" + } + ] + + params = self._construct_test_params(season=2021, round=16, result=16) + + assert self.r.get_statuses(params) == expected + + + def test_get_driver_standings(self): + expected = [ + { + "season":"2021", + "round":"16", + "DriverStandings":[ + { + "position":"10", + "positionText":"10", + "points":"58", + "wins":"0", + "Driver": test_constants.ALONSO, + "Constructors":[ + test_constants.ALPINE + ] + } + ] + } + ] + + params = self._construct_test_params(season=2021, round=16, driver="alonso") + + assert self.r.get_driver_standings(params) == expected + + + def test_get_constructor_standings(self): + expected = [ + { + "season":"2021", + "round":"16", + "ConstructorStandings":[ + { + "position":"5", + "positionText":"5", + "points":"104", + "wins":"1", + "Constructor": test_constants.ALPINE + } + ] + } + ] + + params = self._construct_test_params(season=2021, round=16, standing=5) + + assert self.r.get_constructor_standings(params) == expected + + + def test_get_laps(self): + expected = [ + { + "season":"2008", + "round":"5", + "url":"http://en.wikipedia.org/wiki/2008_Turkish_Grand_Prix", + "raceName":"Turkish Grand Prix", + "Circuit": test_constants.ISTANBUL, + "date":"2008-05-11", + "time":"12:00:00Z", + "Laps":[ + { + "number":"1", + "Timings":[ + { + "driverId":"alonso", + "position":"5", + "time":"1:57.681" + } + ] + } + ] + } + ] + + params = self._construct_test_params(season=2008, round=5, driver="alonso", lap=1) + + assert self.r.get_laps(params) == expected + + + def test_get_pit_stops(self): + expected = [ + { + "season":"2021", + "round":"16", + "url":"http://en.wikipedia.org/wiki/2021_Turkish_Grand_Prix", + "raceName":"Turkish Grand Prix", + "Circuit": test_constants.ISTANBUL, + "date":"2021-10-10", + "time":"12:00:00Z", + "PitStops":[ + { + "driverId":"alonso", + "lap":"30", + "stop":"1", + "time":"15:51:43", + "duration":"29.116" + } + ] + } + ] + + params = self._construct_test_params(season=2021, round=16, driver="alonso", pit_stop=1) + + assert self.r.get_pit_stops(params) == expected \ No newline at end of file