Fix testTensorOpScale test to improve coverage.