def diagOp(self, diag, dtype, expected_ans):
    self._diagOp(diag, dtype, expected_ans, False)
    self._diagOp(diag, dtype, expected_ans, True)