From 7e322e244617b1f91b41e1da6e56d08c63aefaf5 Mon Sep 17 00:00:00 2001 From: Pim Nelissen Date: Fri, 30 Jan 2026 18:57:33 +0100 Subject: [PATCH] Add test for testing seed fixing in RandomRoadGenerator --- pyproject.toml | 5 ++++- tests/test_generators.py | 26 ++++++++++++++++++++++++++ 2 files changed, 30 insertions(+), 1 deletion(-) create mode 100644 tests/test_generators.py diff --git a/pyproject.toml b/pyproject.toml index eecd68e..0981831 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -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" \ No newline at end of file +Issues = "https://github.com/pim-n/road-gen/issues" + +[project.optional-dependencies] +dev = ["pytest"] \ No newline at end of file diff --git a/tests/test_generators.py b/tests/test_generators.py new file mode 100644 index 0000000..8dae427 --- /dev/null +++ b/tests/test_generators.py @@ -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) + \ No newline at end of file