tests/conftest.py

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
# SPDX-License-Identifier: AGPL-3.0-only
# Copyright (C) 2026 SWGY, Inc.
from __future__ import annotations

import matplotlib
import numpy as np
import pytest
from numpy.random import Generator

from iohmm_evac.params import SimulationConfig

matplotlib.use("Agg")


@pytest.fixture
def rng() -> Generator:
    return np.random.default_rng(0)


@pytest.fixture
def small_config() -> SimulationConfig:
    return SimulationConfig(n_households=200, n_hours=24, seed=0)