Commit 05e3cb74 authored by Philipp Arras's avatar Philipp Arras
Browse files

fixup! Remove jax correlated field again

parent e67874df
Pipeline #104719 canceled with stages
in 22 minutes and 37 seconds
......@@ -80,25 +80,3 @@ def test_jax_energy():
continue
pos1 = ift.from_random(e.domain)
ift.extra.assert_allclose(e0(lin).metric(pos1), e(lin).metric(pos1))
def test_cf():
dom = ift.RGSpace([13, 14], distances=(0.89, 0.9))
args = {
'offset_mean': 0,
'offset_std': None,
'fluctuations': (2.1, 0.8),
'loglogavgslope': (-2., 1.2),
'flexibility': None,
'asperity': None,
}
correlated_field = ift.SimpleCorrelatedField(dom, **args)
correlated_field_1 = ift.SimpleCorrelatedField(dom, **args, jax=True)
loc = ift.from_random(correlated_field.domain)
ift.extra.assert_allclose(correlated_field(loc), correlated_field_1(loc))
pspec = correlated_field.power_spectrum
pspec_1 = correlated_field_1.power_spectrum
ift.extra.assert_allclose(pspec.force(loc), pspec_1.force(loc))
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment