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))