physics/pdme
2
0
Fork 0
Code Issues Pull Requests 9 Projects Releases Wiki Activity

Adds all the scripts

Browse Source
This commit is contained in:
Deepak Mallubhotla
2022-01-10 09:41:06 -06:00
parent 72cd281099
commit 778361bece
44 changed files with 3003 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
tests/model/test_fixed_z_discretization.py
View File

@@ -4,11 +4,12 @@ import numpy
def test_fixed_z_plane_model_discretization():
model = FixedZPlaneModel(4, -10, 10, -10, 10, 1)
discretisation = FixedZPlaneDiscretisation(model, 2, 5, 15)
discretisation = FixedZPlaneDiscretisation(model, 1, 2, 5, 15)
# x: (-10, 0) and (0, 10)
# y: (-10, -6, -2, 2, 6, 10)
assert discretisation.cell_count == 10
assert discretisation.pz_step == 30
assert discretisation.x_step == 10
assert discretisation.y_step == 4
numpy.testing.assert_allclose(discretisation.bounds((0, 0)), ((-15, -10, -10, -numpy.inf), (15, 0, -6, numpy.inf)))
numpy.testing.assert_allclose(list(discretisation.all_indices()), list(numpy.ndindex((2, 5))))
numpy.testing.assert_allclose(discretisation.bounds((0, 0, 0)), ((-15, -10, -10, -numpy.inf), (15, 0, -6, numpy.inf)))
numpy.testing.assert_allclose(list(discretisation.all_indices()), list(numpy.ndindex((1, 2, 5))))

9
tests/model/test_unrestricted_discretization.py
View File

@@ -4,12 +4,15 @@ import numpy
def test_unrestricted_model_discretization():
model = UnrestrictedModel(-10, 10, -10, 10, -10, 10, 1)
discretisation = UnrestrictedDiscretisation(model, 2, 5, 1, 15)
discretisation = UnrestrictedDiscretisation(model, 1, 1, 2, 2, 5, 1, 15)
# x: (-10, 0) and (0, 10)
# y: (-10, -6, -2, 2, 6, 10)
assert discretisation.cell_count == 10
assert discretisation.px_step == 30
assert discretisation.py_step == 30
assert discretisation.pz_step == 15
assert discretisation.x_step == 10
assert discretisation.y_step == 4
assert discretisation.z_step == 20
numpy.testing.assert_allclose(discretisation.bounds((0, 0, 0)), ((-15, -15, -15, -10, -10, -10, -numpy.inf), (15, 15, 15, 0, -6, 10, numpy.inf)))
numpy.testing.assert_allclose(list(discretisation.all_indices()), list(numpy.ndindex((2, 5, 1))))
numpy.testing.assert_allclose(discretisation.bounds((0, 0, 0, 0, 0, 0)), ((-15, -15, -15, -10, -10, -10, -numpy.inf), (15, 15, 0, 0, -6, 10, numpy.inf)))
numpy.testing.assert_allclose(list(discretisation.all_indices()), list(numpy.ndindex((1, 1, 2, 2, 5, 1))))