add tests for feature count and probability