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')