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)
