diff --git a/core/src/test/scala/com/salesforce/op/stages/impl/insights/RecordInsightsLOCOTest.scala b/core/src/test/scala/com/salesforce/op/stages/impl/insights/RecordInsightsLOCOTest.scala index c6178bb584..3ac59bab28 100644 --- a/core/src/test/scala/com/salesforce/op/stages/impl/insights/RecordInsightsLOCOTest.scala +++ b/core/src/test/scala/com/salesforce/op/stages/impl/insights/RecordInsightsLOCOTest.scala @@ -232,8 +232,8 @@ class RecordInsightsLOCOTest extends FunSpec with TestSparkContext with RecordIn info("Each feature vector should only have either three or four non-zero entries. One each from country and " + "picklist, while currency can have either two (if it's null the currency column will be filled with the mean)" + " or just one if it's not null.") - it("should pick between 1 and 4 of the features") { - all(parsed.map(_.size)) should (be >= 1 and be <= 4) + it("should pick between 0 and 4 features") { + all(parsed.map(_.size)) should (be >= 0 and be <= 4) } // Grab the feature vector metadata for comparison against the LOCO record insights