added unit test for restoration of collapse class