def test_2mux(self):
mux = Mux(2)
mux.input_1 = 1.5
mux.input_2 = "a"
mux.execute()
self.assertEqual(mux.output, [1.5, "a"])
def test_change_n_mux(self):
mux = Mux(2)
mux.input_1 = 1.5
mux.input_2 = "a"
mux.n = 3
#check that old traits were removed
self.assertEqual(mux.input_1, None)
self.assertEqual(mux.input_2, None)
self.assertEqual(mux.input_3, None)
