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)