timmens/causal-forest

Add unit tests.

timmens opened this issue · 2 comments

Add unit tests to all functions in causaltree.py.

I added unit tests for functions

  1. _compute_valid_splitting_indices
  2. _retrieve_index
  3. _predict_row_causaltree

For now it would be especially important to add unit tests for functions located in cforest/forest.py.