TST: Add unit tests for Categorical node