mirror of
https://github.com/ImperialCollegeLondon/ReCoDE_MCMCFF.git
synced 2025-06-26 08:51:16 +02:00
18 lines
426 B
Python
18 lines
426 B
Python
import numpy as np
|
|
from hypothesis import given
|
|
from hypothesis import strategies as st
|
|
from hypothesis.extra import numpy as hnp
|
|
|
|
from MCFF.ising_model import energy, energy_numpy
|
|
|
|
|
|
@given(
|
|
hnp.arrays(
|
|
dtype=int,
|
|
shape=hnp.array_shapes(min_dims=2, max_dims=2),
|
|
elements=st.sampled_from([1, -1]),
|
|
)
|
|
)
|
|
def test_generated_states(state):
|
|
assert np.allclose(energy(state), energy_numpy(state))
|