def test_ground_state(): ground = Wavefunction.zeros(2) assert len(ground) == 2 assert ground.amplitudes[0] == 1.0
def test_imag_wavefunction_io(): wf = Wavefunction([0, 1j, 0, 0, 0, 0, 0, 0]) save_wavefunction(wf, "wavefunction.json") loaded_wf = load_wavefunction("wavefunction.json") assert np.allclose(wf.amplitudes, loaded_wf.amplitudes) remove_file_if_exists("wavefunction.json")
def wvf(): amps = np.array([1.0, 1.0j, 0.000005, 0.02]) amps /= np.sqrt(np.sum(np.abs(amps)**2)) return Wavefunction(amps)
def test_imag_wavefunction_io(self): wf = Wavefunction([0, 1j, 0, 0, 0, 0, 0, 0]) save_wavefunction(wf, "wavefunction.json") loaded_wf = load_wavefunction("wavefunction.json") self.assertTrue(np.allclose(wf.amplitudes, loaded_wf.amplitudes)) os.remove("wavefunction.json")
def test_real_wavefunction_io(self): wf = Wavefunction([0, 1, 0, 0, 0, 0, 0, 0]) save_wavefunction(wf, 'wavefunction.json') loaded_wf = load_wavefunction('wavefunction.json') self.assertTrue(np.allclose(wf.amplitudes, loaded_wf.amplitudes)) os.remove('wavefunction.json')
def get_wavefunction(self, circuit): super().get_wavefunction(circuit) qulacs_state = self.get_qulacs_state_from_circuit(circuit) amplitudes = qulacs_state.get_vector() return Wavefunction(amplitudes)
def wvf(): return Wavefunction(np.array([1.0, 1.j, 0.000005, 0.02]))
def test_ground_state(): ground = Wavefunction.ground(2) assert len(ground) == 2 assert ground[0] == 1.0