def test_nonsensical_units(self):
F = 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)
q = 1 * u.C
with pytest.raises(u.UnitTypeError):
drifts.force_drift(F, B, q)
def test_force_x_B_1d_arrays(self):
F = u.Quantity([1, 0, 0], unit=u.N)
B = u.Quantity([0, 1, 0], unit=u.T)
q = 1 * u.C
result = drifts.force_drift(2 * F, 3 * B, q)
assert_quantity_allclose(result,
(2 / 3) * u.Quantity([0, 0, 1], u.m / u.s))
