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)