refactor test to include mock visualizations only for tests that require it