mirror of
https://github.com/pim-n/road-gen.git
synced 2026-02-03 09:23:09 +01:00
Add test for testing seed fixing in RandomRoadGenerator
This commit is contained in:
@ -27,3 +27,6 @@ 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"
|
||||
|
||||
[project.optional-dependencies]
|
||||
dev = ["pytest"]
|
||||
26
tests/test_generators.py
Normal file
26
tests/test_generators.py
Normal 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)
|
||||
|
||||
Reference in New Issue
Block a user