def rotate_psi(self, basis, unitary_dict, vis):
return ts.rotate_psi(self.nn_state, basis, vis, unitary_dict)
def test_rotate_psi(num_visible, wvfn_type):
nn_state = wvfn_type(num_visible, gpu=False)
basis = "X" * num_visible
unitary_dict = create_dict()
rotate_psi(nn_state, basis, nn_state.generate_hilbert_space(), unitary_dict)