Add tests for visitors

Add unit tests testing the functionality of visitors.