Skip to content

Commit

Permalink
update tests
Browse files Browse the repository at this point in the history
  • Loading branch information
jvshields committed Jan 17, 2024
1 parent 9ff7861 commit 1bfed14
Showing 1 changed file with 4 additions and 4 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -253,13 +253,13 @@ def test_calc_n_effective_cuda_wrapped_sample_cuda_values(
(
1,
0,
(0.51 * 0.642) ** (-3 / 2),
(0.60 * 0.642) ** (-3 / 2),
1.0,
),
(
np.array(2 * [1]),
np.array(2 * [0]),
np.array(2 * [(0.51 * 0.642) ** (-3 / 2)]),
np.array(2 * [(0.60 * 0.642) ** (-3 / 2)]),
np.array(2 * [1.0]),
),
],
Expand Down Expand Up @@ -289,7 +289,7 @@ def test_calc_gamma_linear_stark_sample_values(
(
np.array(2 * [1]),
np.array(2 * [0]),
np.array(2 * [(0.51 * 0.642) ** (-3 / 2)]),
np.array(2 * [(0.60 * 0.642) ** (-3 / 2)]),
np.array(2 * [1.0]),
),
],
Expand Down Expand Up @@ -330,7 +330,7 @@ def test_calc_gamma_linear_stark_cuda_unwrapped_sample_values(
(
np.array(2 * [1]),
np.array(2 * [0]),
np.array(2 * [(0.51 * 0.642) ** (-3 / 2)]),
np.array(2 * [(0.60 * 0.642) ** (-3 / 2)]),
np.array(2 * [1.0]),
),
],
Expand Down

0 comments on commit 1bfed14

Please sign in to comment.