Add function to erase state to test switch driver

Call this function at the beginning of each test