Add unit tests for the _get_tire_loads function