Skip to content

Commit

Permalink
[skip ci] update Polar3DVector tests
Browse files Browse the repository at this point in the history
  • Loading branch information
@ianna
ianna committed May 11, 2021
1 parent 46a440a commit c08f109
Show file tree
Hide file tree
Showing 3 changed files with 412 additions and 72 deletions.
126 changes: 82 additions & 44 deletions tests/root/test_Polar2DVector.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -213,8 +213,15 @@ def test_Rotate(constructor, angle, coordinates):
vector.obj(**dict(zip(["rho", "phi"], constructor))), coordinates
)().rotateZ(angle)
res_vec = vec.rotateZ(angle)
assert ref_vec.R() == pytest.approx(res_vec.rho) and ref_vec.Phi() == pytest.approx(
res_vec.phi
assert ref_vec.R() == pytest.approx(
res_vec.rho,
1.0e-6,
1.0e-6,
)
assert ref_vec.Phi() == pytest.approx(
res_vec.phi,
1.0e-6,
1.0e-6,
)


Expand All @@ -236,8 +243,15 @@ def test_fuzz_Rotate(constructor, angle, coordinates):
ref_vec.Rotate(angle)
vec = getattr(vector.obj(**dict(zip(["rho", "phi"], constructor))), coordinates)()
res_vec = vec.rotateZ(angle)
assert ref_vec.R() == pytest.approx(res_vec.rho) and ref_vec.Phi() == pytest.approx(
res_vec.phi
assert ref_vec.R() == pytest.approx(
res_vec.rho,
1.0e-6,
1.0e-6,
)
assert ref_vec.Phi() == pytest.approx(
res_vec.phi,
1.0e-6,
1.0e-6,
)


Expand All @@ -247,9 +261,8 @@ def test_Unit(constructor, coordinates):
ref_vec = ROOT.Math.Polar2DVector(*constructor).Unit()
vec = getattr(vector.obj(**dict(zip(["rho", "phi"], constructor))), coordinates)()
res_vec = vec.unit
assert ref_vec.R() == pytest.approx(
res_vec().rho
) and ref_vec.Phi() == pytest.approx(res_vec().phi)
assert ref_vec.R() == pytest.approx(res_vec().rho)
assert ref_vec.Phi() == pytest.approx(res_vec().phi)


# Run the same tests within hypothesis
Expand All @@ -267,9 +280,8 @@ def test_fuzz_Unit(constructor, coordinates):
ref_vec = ROOT.Math.Polar2DVector(*constructor).Unit()
vec = getattr(vector.obj(**dict(zip(["rho", "phi"], constructor))), coordinates)()
res_vec = vec.unit
assert ref_vec.R() == pytest.approx(
res_vec().rho
) and ref_vec.Phi() == pytest.approx(res_vec().phi)
assert ref_vec.R() == pytest.approx(res_vec().rho)
assert ref_vec.Phi() == pytest.approx(res_vec().phi)


# Run a test that compares ROOT's 'X()' and 'Y()' with vector's 'x' and 'y' for all cases.
Expand Down Expand Up @@ -308,8 +320,15 @@ def test_add(constructor, coordinates):
)().add(
getattr(vector.obj(**dict(zip(["rho", "phi"], constructor))), coordinates)()
)
assert ref_vec.R() == pytest.approx(vec.rho) and ref_vec.Phi() == pytest.approx(
vec.phi
assert ref_vec.R() == pytest.approx(
vec.rho,
1.0e-6,
1.0e-6,
)
assert ref_vec.Phi() == pytest.approx(
vec.phi,
1.0e-6,
1.0e-6,
)


Expand Down Expand Up @@ -341,8 +360,15 @@ def test_fuzz_add(constructor1, constructor2, coordinates):
)().add(
getattr(vector.obj(**dict(zip(["rho", "phi"], constructor2))), coordinates)()
)
assert ref_vec.R() == pytest.approx(vec.rho) and ref_vec.Phi() == pytest.approx(
vec.phi
assert ref_vec.R() == pytest.approx(
vec.rho,
1.0e-6,
1.0e-6,
)
assert ref_vec.Phi() == pytest.approx(
vec.phi,
1.0e-6,
1.0e-6,
)


Expand All @@ -355,8 +381,15 @@ def test_sub(constructor, coordinates):
vec1 = getattr(vector.obj(**dict(zip(["rho", "phi"], constructor))), coordinates)()
vec2 = getattr(vector.obj(**dict(zip(["rho", "phi"], constructor))), coordinates)()
res_vec = vec1.subtract(vec2)
assert ref_vec.R() == pytest.approx(res_vec.rho) and ref_vec.Phi() == pytest.approx(
res_vec.phi
assert ref_vec.R() == pytest.approx(
res_vec.rho,
1.0e-6,
1.0e-6,
)
assert ref_vec.Phi() == pytest.approx(
res_vec.phi,
1.0e-6,
1.0e-6,
)


Expand Down Expand Up @@ -386,8 +419,15 @@ def test_fuzz_sub(constructor1, constructor2, coordinates):
vec1 = getattr(vector.obj(**dict(zip(["rho", "phi"], constructor1))), coordinates)()
vec2 = getattr(vector.obj(**dict(zip(["rho", "phi"], constructor2))), coordinates)()
res_vec = vec1.subtract(vec2)
assert ref_vec.R() == pytest.approx(res_vec.rho) and ref_vec.Phi() == pytest.approx(
res_vec.phi
assert ref_vec.R() == pytest.approx(
res_vec.rho,
1.0e-6,
1.0e-6,
)
assert ref_vec.Phi() == pytest.approx(
res_vec.phi,
1.0e-6,
1.0e-6,
)


Expand All @@ -398,9 +438,8 @@ def test_neg(constructor, coordinates):
vec = getattr(
vector.obj(**dict(zip(["rho", "phi"], constructor))), coordinates
)().__neg__
assert ref_vec.R() == pytest.approx(vec().rho) and ref_vec.Phi() == pytest.approx(
vec().phi
)
assert ref_vec.R() == pytest.approx(vec().rho)
assert ref_vec.Phi() == pytest.approx(vec().phi)


# Run the same tests within hypothesis
Expand All @@ -419,9 +458,8 @@ def test_fuzz_neg(constructor, coordinates):
vec = getattr(
vector.obj(**dict(zip(["rho", "phi"], constructor))), coordinates
)().__neg__
assert ref_vec.R() == pytest.approx(vec().rho) and ref_vec.Phi() == pytest.approx(
vec().phi
)
assert ref_vec.R() == pytest.approx(vec().rho)
assert ref_vec.Phi() == pytest.approx(vec().phi)


# Run a test that compares ROOT's '__mul__' with vector's 'mul' for all cases.
Expand All @@ -431,9 +469,8 @@ def test_mul(constructor, scalar, coordinates):
vec = getattr(
vector.obj(**dict(zip(["rho", "phi"], constructor))), coordinates
)().__mul__(scalar)
assert ref_vec.R() == pytest.approx(vec.rho) and ref_vec.Phi() == pytest.approx(
vec.phi
)
assert ref_vec.R() == pytest.approx(vec.rho)
assert ref_vec.Phi() == pytest.approx(vec.phi)


# Run the same tests within hypothesis
Expand All @@ -454,21 +491,21 @@ def test_fuzz_mul(constructor, scalar, coordinates):
vec = getattr(
vector.obj(**dict(zip(["rho", "phi"], constructor))), coordinates
)().__mul__(scalar)
assert ref_vec.R() == pytest.approx(vec.rho) and ref_vec.Phi() == pytest.approx(
vec.phi
)
assert ref_vec.R() == pytest.approx(vec.rho)
assert ref_vec.Phi() == pytest.approx(vec.phi)


# Run a test that compares ROOT's '__truediv__' with vector's '__truediv__' for all cases.
@pytest.mark.parametrize("constructor", constructor)
def test_truediv(constructor, scalar, coordinates):
ref_vec = ROOT.Math.Polar2DVector(*constructor).__truediv__(scalar)
vec = getattr(
vector.obj(**dict(zip(["rho", "phi"], constructor))), coordinates
)().__truediv__(scalar)
assert ref_vec.R() == pytest.approx(vec.rho) and ref_vec.Phi() == pytest.approx(
vec.phi
)
# FIXME:
if scalar != 0:
ref_vec = ROOT.Math.Polar2DVector(*constructor).__truediv__(scalar)
vec = getattr(
vector.obj(**dict(zip(["rho", "phi"], constructor))), coordinates
)().__truediv__(scalar)
assert ref_vec.R() == pytest.approx(vec.rho)
assert ref_vec.Phi() == pytest.approx(vec.phi)


# Run the same tests within hypothesis
Expand All @@ -485,13 +522,14 @@ def test_truediv(constructor, scalar, coordinates):
| st.integers(min_value=-10e7, max_value=10e7),
)
def test_fuzz_truediv(constructor, scalar, coordinates):
ref_vec = ROOT.Math.Polar2DVector(*constructor).__truediv__(scalar)
vec = getattr(
vector.obj(**dict(zip(["rho", "phi"], constructor))), coordinates
)().__truediv__(scalar)
assert ref_vec.R() == pytest.approx(vec.rho) and ref_vec.Phi() == pytest.approx(
vec.phi
)
# FIXME:
if scalar != 0:
ref_vec = ROOT.Math.Polar2DVector(*constructor).__truediv__(scalar)
vec = getattr(
vector.obj(**dict(zip(["rho", "phi"], constructor))), coordinates
)().__truediv__(scalar)
assert ref_vec.R() == pytest.approx(vec.rho)
assert ref_vec.Phi() == pytest.approx(vec.phi)


# Run a test that compares ROOT's '__eq__' with vector's 'isclose' for all cases.
Expand Down
Loading

0 comments on commit c08f109

Please sign in to comment.