def test_Plasma3D_derived_vars():
r"""Function to test derived variables of the Plasma3D class.
Tests the shapes, units and values of variables derived from core
variables. The core variables are set with arbitrary uniform
values.
"""
test_plasma = plasma3d.Plasma3D(domain_x=np.linspace(0, 1, 64) * u.m,
domain_y=np.linspace(0, 1, 64) * u.m,
domain_z=np.linspace(0, 1, 1) * u.m)
# Set an arbitrary uniform values throughout the plasma
test_plasma.density[...] = 2.0 * u.kg / u.m ** 3
test_plasma.momentum[...] = 10.0 * u.kg / (u.m ** 2 * u.s)
test_plasma.pressure[...] = 1 * u.Pa
test_plasma.magnetic_field[...] = 0.01 * u.T
test_plasma.electric_field[...] = 0.01 * u.V / u.m
# Test derived variable units and shapes
assert test_plasma.velocity.shape == test_plasma.momentum.shape
assert (test_plasma.velocity == 5.0 * u.m / u.s).all()
assert test_plasma.magnetic_field_strength.shape == \
test_plasma.magnetic_field.shape[1:]
assert test_plasma.magnetic_field_strength.si.unit == u.T
assert np.allclose(test_plasma.magnetic_field_strength.value, 0.017320508)
assert test_plasma.electric_field_strength.shape == \
test_plasma.electric_field.shape[1:]
assert test_plasma.electric_field_strength.si.unit == u.V / u.m
assert test_plasma.alfven_speed.shape == test_plasma.density.shape
assert test_plasma.alfven_speed.unit.si == u.m / u.s
assert np.allclose(test_plasma.alfven_speed.value, 10.92548431)
def test_Plasma3D_setup(grid_dimensions, expected_size):
r"""Function to test basic setup of the Plasma3D object.
Tests that a Plasma3D object initiated with a particular
specification behaves in the correct way.
Parameters
----------
grid_dimensions : tuple of ints
Grid size of the Plasma3D object to test. Must be a tuple of
length 3, indicating length of the grid in x, y, and z
directions respectively. Directions not needed should have a
length of 1.
expected_size : int
Product of grid dimensions.
Examples
--------
>>> test_Plasma3D_setup((10, 10, 10), 1000)
>>> test_Plasma3D_setup((100, 10, 1), 1000)
"""
x, y, z = grid_dimensions
test_plasma = plasma3d.Plasma3D(
domain_x=np.linspace(0, 1, x) * u.m,
domain_y=np.linspace(0, 1, y) * u.m,
domain_z=np.linspace(0, 1, z) * u.m,
)
# Basic grid setup
assert test_plasma.x.size == x
assert test_plasma.y.size == y
assert test_plasma.z.size == z
assert test_plasma.grid.size == 3 * expected_size
# Core variable units and shapes
assert test_plasma.density.size == expected_size
assert test_plasma.density.si.unit == u.kg / u.m**3
assert test_plasma.momentum.size == 3 * expected_size
assert test_plasma.momentum.si.unit == u.kg / (u.m**2 * u.s)
assert test_plasma.pressure.size == expected_size
assert test_plasma.pressure.si.unit == u.Pa
assert test_plasma.magnetic_field.size == 3 * expected_size
assert test_plasma.magnetic_field.si.unit == u.T
assert test_plasma.electric_field.size == 3 * expected_size
assert test_plasma.electric_field.si.unit == u.V / u.m
def test_Plasma3D_add_magnetostatics():
r"""Function to test add_magnetostatic function
"""
dipole = magnetostatics.MagneticDipole(
np.array([0, 0, 1]) * u.A * u.m * u.m,
np.array([0, 0, 0]) * u.m)
cw = magnetostatics.CircularWire(np.array([0, 0, 1]),
np.array([0, 0, 0]) * u.m, 1 * u.m,
1 * u.A)
gw_cw = cw.to_GeneralWire()
iw = magnetostatics.InfiniteStraightWire(np.array([0, 1, 0]),
np.array([0, 0, 0]) * u.m,
1 * u.A)
plasma = plasma3d.Plasma3D(domain_x=np.linspace(-2, 2, 30) * u.m,
domain_y=np.linspace(0, 0, 1) * u.m,
domain_z=np.linspace(-2, 2, 20) * u.m)
plasma.add_magnetostatic(dipole, cw, gw_cw, iw)
