Move in-memory driver test to appropriate package