def test_add_inplace(self): """Test inplace add method.""" chan = UnitaryChannel([[1, 2], [3, 4]]) chan.add(chan, inplace=True) self.assertEqual(chan, UnitaryChannel([[2, 4], [6, 8]])) chan = UnitaryChannel([[1, 2], [3, 4]]) chan += chan self.assertEqual(chan, UnitaryChannel([[2, 4], [6, 8]]))
def test_add(self): """Test add method.""" chan = UnitaryChannel([[1, 2], [3, 4]]) self.assertEqual(chan.add(chan), UnitaryChannel([[2, 4], [6, 8]])) self.assertEqual(chan + chan, UnitaryChannel([[2, 4], [6, 8]]))