pratyush/ergast-py
Archived
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Added full basic tests for requester get functions

Browse Source
This commit is contained in:
Samuel Roach 2022-06-03 15:25:45 +01:00
parent dcddfea77e
commit 342a6d69cd
2 changed files with 277 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
tests/test_constants.py
View File

@ -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"
}
}

262
tests/test_requester.py
View File

@ -5,6 +5,8 @@ import tests.test_constants as test_constants
class TestRequester():
"""
Tests for the Requester class
~~~
<Note>
"""
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
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