changed driver code to assert statements