def test_ExB_3d_array(self):
E = u.Quantity([[[1, 0, 0]]], unit=u.V / u.m)
B = u.Quantity([[[0, 1, 0]]], unit=u.T)
result = drifts.ExB_drift(2 * E, 3 * B)
assert_quantity_allclose(result, (2 / 3) *
u.Quantity([[[0, 0, 1]]], unit=u.m / u.s))
def test_nonsensical_units(self):
E = u.Quantity([[1, 0, 0],
[1, 0, 0],
[1, 0, 0]], unit=u.mm)
B = u.Quantity([[0, 1, 0],
[0, 1, 0],
[0, 1, 0]], unit=u.kg)
with pytest.raises(u.UnitTypeError):
drifts.ExB_drift(E, B)
