Updated test driver assertions