Updated decision tree unit test to perform validation