def test_active_modes(self): qubit = Transmon("qubit", levels=3, kerr=-200e-3) cavity = Cavity("cavity", levels=10, kerr=-10e-6) system = System("system", modes=[qubit]) system.active_modes = [qubit] with self.assertRaises(ValueError): system.active_modes = [qubit, cavity] system.modes = [qubit, cavity] system.active_modes = [qubit, cavity] system.modes = [cavity] with self.assertRaises(ValueError): _ = system.active_modes
def test_order_modes(self): qubit = Transmon("qubit", levels=3, kerr=-200e-3) cavity = Cavity("cavity", levels=10, kerr=-10e-6) system = System("system", modes=[qubit, cavity]) system.order_modes = True system.active_modes = [qubit, cavity] self.assertEqual(system.modes, [qubit, cavity]) self.assertEqual(system.active_modes, [qubit, cavity]) system.active_modes = [cavity, qubit] self.assertEqual(system.modes, [qubit, cavity]) self.assertEqual(system.active_modes, [qubit, cavity]) system = System("system", modes=[qubit, cavity]) system.order_modes = False system.active_modes = [qubit, cavity] self.assertEqual(system.modes, [qubit, cavity]) self.assertEqual(system.active_modes, [qubit, cavity]) system.active_modes = [cavity, qubit] self.assertEqual(system.modes, [qubit, cavity]) self.assertEqual(system.active_modes, [cavity, qubit])