def test_resetting_qubits(self):
host = Host('A')
q1 = Qubit(host)
q2 = Qubit(host)
q3 = Qubit(host)
host.add_epr('B', q1)
host.add_data_qubit('B', q2)
host.add_ghz_qubit('B', q3)
qs = host.get_data_qubits('B')
self.assertEqual(len(qs), 1)
host.reset_data_qubits('B')
qs = host.get_data_qubits('B')
self.assertEqual(len(qs), 0)
def test_get_data_qubits(self):
host = Host('A')
q1 = Qubit(host)
q2 = Qubit(host)
q3 = Qubit(host)
host.add_data_qubit('B', q1)
qs = host.get_data_qubits('B')
self.assertEqual(len(qs), 1)
host.add_data_qubit('B', q2)
host.add_data_qubit('B', q3)
qs = host.get_data_qubits('B')
self.assertEqual(len(qs), 3)
qs = host.get_data_qubits('B', remove_from_storage=True)
self.assertEqual(len(qs), 3)
qs = host.get_data_qubits('B', remove_from_storage=True)
self.assertEqual(len(qs), 0)
