converted driver test code to assert statements