Skip to content

Commit

Permalink
examples: size reduction in test_gradientFWI to maintain uniformity a…
Browse files Browse the repository at this point in the history
…nd minimize CI times
  • Loading branch information
ofmla committed Feb 19, 2021
1 parent f520922 commit 2de2bfe
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions tests/test_gradient.py
Original file line number Diff line number Diff line change
Expand Up @@ -140,10 +140,10 @@ def test_gradient_equivalence(self, shape, kernel, space_order, preset, nbl, dty
assert(np.allclose(grad_u.data, grad_uv.data, rtol=tolerance, atol=tolerance))

@pytest.mark.parametrize('dtype, space_order, kernel, shape, ckp, setup_func', [
(np.float32, 4, 'OT2', (70, 80), True, setup),
(np.float32, 4, 'OT2', (70, 80), False, setup),
(np.float64, 4, 'OT2', (70, 80), True, setup),
(np.float64, 4, 'OT2', (70, 80), False, setup),
(np.float32, 4, 'OT2', (50, 60), True, setup),
(np.float32, 4, 'OT2', (50, 60), False, setup),
(np.float64, 4, 'OT2', (50, 60), True, setup),
(np.float64, 4, 'OT2', (50, 60), False, setup),
(np.float32, 4, 'centered', (50, 60), True, tti_setup),
(np.float32, 4, 'centered', (50, 60), False, tti_setup),
(np.float64, 4, 'centered', (50, 60), True, tti_setup),
Expand Down

0 comments on commit 2de2bfe

Please sign in to comment.