Skip to content

Commit

Permalink
Fix: Rounding to closest in tests
Browse files Browse the repository at this point in the history
  • Loading branch information
ashvardanian committed Oct 14, 2024
1 parent 6078973 commit 4bffbc7
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions scripts/test.py
Original file line number Diff line number Diff line change
Expand Up @@ -688,7 +688,7 @@ def test_dense_i8(ndim, metric, stats_fixture):
expected_dt, expected = profile(baseline_kernel, a.astype(np.int64), b.astype(np.int64))
result_dt, result = profile(simd_kernel, a, b)

assert int(result) == int(expected), f"Expected {expected}, but got {result}"
assert round(result) == round(expected), f"Expected {expected}, but got {result}"
collect_errors(metric, ndim, "int8", accurate, accurate_dt, expected, expected_dt, result, result_dt, stats_fixture)

#! Fun fact: SciPy doesn't actually raise an `OverflowError` when overflow happens
Expand Down Expand Up @@ -903,7 +903,7 @@ def test_intersect(dtype, first_length_bound, second_length_bound):
expected = baseline_intersect(a, b)
result = simd.intersect(a, b)

assert int(expected) == int(result), f"Missing {np.intersect1d(a, b)} from {a} and {b}"
assert round(expected) == round(result), f"Missing {np.intersect1d(a, b)} from {a} and {b}"


@pytest.mark.skipif(not numpy_available, reason="NumPy is not installed")
Expand Down

0 comments on commit 4bffbc7

Please sign in to comment.