Deepak
0adca4bcd7
All checks were successful
gitea-physics/pdme/pipeline/head This commit looks good
15 lines
656 B
Python
15 lines
656 B
Python
from pdme.model.fixed_z_plane_model import FixedZPlaneModel, FixedZPlaneDiscretisation
|
|
import numpy
|
|
|
|
|
|
def test_fixed_z_plane_model_cost_and_jac_single():
|
|
model = FixedZPlaneModel(4, -10, 10, -10, 10, 1)
|
|
discretisation = FixedZPlaneDiscretisation(model, 2, 5)
|
|
# x: (-10, 0) and (0, 10)
|
|
# y: (-10, -6, -2, 2, 6, 10)
|
|
assert discretisation.cell_count == 10
|
|
assert discretisation.x_step == 10
|
|
assert discretisation.y_step == 4
|
|
numpy.testing.assert_allclose(discretisation.bounds((0, 0)), ((-numpy.inf, -10, -10, -numpy.inf), (numpy.inf, 0, -6, numpy.inf)))
|
|
numpy.testing.assert_allclose(list(discretisation.all_indices()), list(numpy.ndindex((2, 5))))
|