Add a test for ChiSquare trait