Testcase update for imperative

@aparajit-pratap