Testing PyMC with a Categorical distribution