Add unit test for theme