Added class for visualization frame tests
Modified some tests