from utilities.simulation_utilities import check_inputs

import pytest
import numpy as np


@pytest.mark.parametrize("input,expected", [
    (None, np.ndarray([0])),
    ([0], np.array([0])),
    (1, np.array([1])),
    (range(5), np.array([0,1,2,3,4]))
])
def test_check_inputs(input, expected):
    assert np.allclose(check_inputs(input), expected)
Add a test for check_inputs.
Add a test for check_inputs.
