def test_change_initial_state(self):
    sampler = gibbs.SysGibbsAlgo(num_dim=2)
    sampler.change_initial_state(new_state=np.array([2, 0]))
    npt.assert_array_equal(sampler.initial_state, np.array([2, 0]))
    with self.assertRaises(ValueError):
        sampler.change_initial_state(new_state=np.array([[1], [2]]))
    with self.assertRaises(ValueError):
        sampler.change_initial_state(new_state=np.array([1, 2, 0]))
def test_change_initial_state(self):
    sampler = gibbs.SysGibbsAlgo(num_dim=2)
    sampler.change_initial_state(new_state=np.array([2, 0]))
    npt.assert_array_equal(sampler.initial_state, np.array([2, 0]))
    with self.assertRaises(ValueError):
        sampler.change_initial_state(new_state=np.array([[1], [2]]))
    with self.assertRaises(ValueError):
        sampler.change_initial_state(new_state=np.array([1, 2, 0]))