def test_init(self, j): """Test {j}-qubit automatic dims.""" dim = 2**j op = ScalarOp(dim) self.assertEqual(op.dim, (dim, dim)) self.assertEqual(op.input_dims(), j * (2, )) self.assertEqual(op.output_dims(), j * (2, )) self.assertEqual(op.coeff, 1)
def test_custom_dims(self): """Test custom dims.""" dims = (2, 3, 4, 5) dim = np.product(dims) op = ScalarOp(dims) self.assertEqual(op.dim, (dim, dim)) self.assertEqual(op.input_dims(), dims) self.assertEqual(op.output_dims(), dims) self.assertEqual(op.coeff, 1)