def test_main_algorithm():
import numpy as np
from numpy import testing
from unittest.mock import Mock
density = [1, 1, 1, 1, 1]
energy = MagicMock()
mc = MonteCarlo(energy, density, itermax=5)
acceptance = [True, True, True, True, True]
mc.accept_change = Mock(side_effect=acceptance)
mc.random_agent = Mock(side_effect=[0, 1, 2, 3, 4])
mc.random_direction = Mock(side_effect=[1, 1, 1, 1, -1])
np.testing.assert_equal(mc.step()[1], [0, 1, 1, 2, 1])
