def test_generate_samples(self, results, indices): """Tests that the generate_samples function of HQSDevice provides samples in the correct format expected by PennyLane.""" dev = HQSDevice(3, machine=DUMMY_MACHINE, shots=10, user_email=DUMMY_EMAIL) dev._results = results res = dev.generate_samples() expected_array = np.stack([np.ravel(indices)] * 10) assert res.shape == (dev.shots, dev.num_wires) assert np.all(res == expected_array)
def test_reset(self): """Tests that the ``reset`` method corretly resets data.""" dev = HQSDevice(3, shots=10, machine=DUMMY_MACHINE, user_email=DUMMY_EMAIL) dev._results = ["00"] * 10 dev._samples = np.zeros((10, 3)) dev.shots = 11 dev.reset() assert dev._results is None assert dev._samples is None assert dev.shots == 11 # should not be reset