diff --git a/src/DataFrame-Tests/DataFrameTest.class.st b/src/DataFrame-Tests/DataFrameTest.class.st index f45dc9fb..a1591514 100644 --- a/src/DataFrame-Tests/DataFrameTest.class.st +++ b/src/DataFrame-Tests/DataFrameTest.class.st @@ -1952,6 +1952,61 @@ DataFrameTest >> testHasNilsByColumn [ self assert: df hasNilsByColumn equals: expected ] +{ #category : #'tests - find/select' } +DataFrameTest >> testIncludes [ + + | dataSeries1 dataSeries2 | + dataSeries1 := DataSeries + withKeys: #( City Population BeenThere ) + values: #( London 8.788 false ) + name: 'C'. + dataSeries2 := DataSeries + withKeys: #( City Population BeenThere ) + values: #( Paris 8.788 false ) + name: 'B'. + + self assert: (df includes: dataSeries1). + self deny: (df includes: dataSeries2) +] + +{ #category : #'tests - find/select' } +DataFrameTest >> testIncludesAll [ + + | dataSeries1 dataSeries2 | + dataSeries1 := DataSeries + withKeys: #( City Population BeenThere ) + values: #( London 8.788 false ) + name: 'C'. + dataSeries2 := DataSeries + withKeys: #( City Population BeenThere ) + values: #( Paris 8.788 false ) + name: 'B'. + + self assert: (df includesAll: { dataSeries1 }). + self deny: (df includesAll: { + dataSeries1. + dataSeries2 }) +] + +{ #category : #'tests - find/select' } +DataFrameTest >> testIncludesAny [ + + | dataSeries1 dataSeries2 | + dataSeries1 := DataSeries + withKeys: #( City Population BeenThere ) + values: #( London 8.788 false ) + name: 'C'. + dataSeries2 := DataSeries + withKeys: #( City Population BeenThere ) + values: #( Paris 8.788 false ) + name: 'B'. + + self assert: (df includesAny: { + dataSeries1. + dataSeries2 }). + self deny: (df includesAny: { dataSeries2 }) +] + { #category : #tests } DataFrameTest >> testIndexOfColumnNamed [ | expected actual |