def test_get_xdata(self): """ Tests that the discriminator properly retrieves the x data from the Qiskit result. """ discriminator = LinearIQDiscriminator(self.cal_results, self.qubits, ['00', '11']) xdata = discriminator.get_xdata(self.cal_results) self.assertEqual(len(xdata), self.shots * 2) self.assertEqual(len(xdata[0]), len(self.qubits) * 2) xdata = discriminator.get_xdata(self.cal_results, ['cal_00']) self.assertEqual(len(xdata), self.shots) self.assertEqual(len(xdata[0]), 4)
def test_get_ydata(self): """ Tests that the discriminator properly retrieves the y data from the Qiskit calibration results. """ discriminator = LinearIQDiscriminator(self.cal_results, self.qubits, ['00', '11']) xdata = discriminator.get_xdata(self.cal_results) ydata = discriminator.get_ydata(self.cal_results) self.assertEqual(len(xdata), len(ydata)) ydata = discriminator.get_ydata(self.cal_results, ['cal_00']) self.assertEqual(len(ydata), self.shots) self.assertEqual(ydata[0], '00')