Add test for category class.