tests/test_resource_mappings.py

""" Tests for resource mappings """

import os
import json


# Load in driver names and numbers
resource_dir = os.path.join(os.path.dirname(__file__), "..", "assets")
driver_names = json.load(open(os.path.join(resource_dir, "driver_names.json")))
driver_numbers = json.load(open(os.path.join(resource_dir, "driver_numbers.json")))

for name in driver_names["drivers"]:
    number = driver_numbers.get(name, None)
    if number is None:
        raise ValueError(f"Driver {name} does not have a driver number")
print("All driver names and numbers are mapped")