Add test for testing seed fixing in RandomRoadGenerator

This commit is contained in:
Pim Nelissen
2026-01-30 18:57:33 +01:00
parent f95babc369
commit 7e322e2446
2 changed files with 30 additions and 1 deletions

View File

@ -26,4 +26,7 @@ road-gen = "road_gen.main:main"
[project.urls]
Homepage = "https://github.com/pim-n/road-gen"
Issues = "https://github.com/pim-n/road-gen/issues"
Issues = "https://github.com/pim-n/road-gen/issues"
[project.optional-dependencies]
dev = ["pytest"]

26
tests/test_generators.py Normal file
View File

@ -0,0 +1,26 @@
import numpy as np
import pytest
from road_gen.generators.random_road_generator import RandomRoadGenerator
@pytest.fixture
def test_params():
length = 1_000
ds = 10
velocity = 10
return length, ds, velocity
def test_random_road_generator(test_params):
"""Test whether fixing the seed for RandomRoadGenerator produces identical output."""
generator_1 = RandomRoadGenerator(*test_params)
x1, y1 = generator_1.generate()
generator_2 = RandomRoadGenerator(seed = generator_1.seed, *test_params)
x2, y2 = generator_2.generate()
assert np.array_equal(x1, x2)
assert np.array_equal(y1, y2)