Skip to content

Commit 3bec93e

Browse files
Added DataFrame>> #includes: with tests
1 parent 023a2a6 commit 3bec93e

2 files changed

Lines changed: 24 additions & 0 deletions

File tree

src/DataFrame-Tests/DataFrameTest.class.st

Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1937,6 +1937,23 @@ DataFrameTest >> testHasNilsByColumn [
19371937
self assert: df hasNilsByColumn equals: expected
19381938
]
19391939

1940+
{ #category : #'find-select' }
1941+
DataFrameTest >> testIncludes [
1942+
1943+
| dataSeries1 dataSeries2 |
1944+
dataSeries1 := DataSeries
1945+
withKeys: #( City Population BeenThere )
1946+
values: #( London 8.788 false )
1947+
name: 'C'.
1948+
dataSeries2 := DataSeries
1949+
withKeys: #( City Population BeenThere )
1950+
values: #( Paris 8.788 false )
1951+
name: 'B'.
1952+
1953+
self assert: (df includes: dataSeries1).
1954+
self deny: (df includes: dataSeries2)
1955+
]
1956+
19401957
{ #category : #tests }
19411958
DataFrameTest >> testIndexOfColumnNamed [
19421959
| expected actual |

src/DataFrame/DataFrame.class.st

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1243,6 +1243,13 @@ DataFrame >> head: aNumber [
12431243
^ self rowsAt: (1 to: (self numberOfRows min: aNumber))
12441244
]
12451245

1246+
{ #category : #'find-select' }
1247+
DataFrame >> includes: aDataSeries [
1248+
"Returns true if the DataFrame includes the specified DataSeries."
1249+
1250+
^ self rows anySatisfy: [ :row | row = aDataSeries ]
1251+
]
1252+
12461253
{ #category : #accessing }
12471254
DataFrame >> indexOfColumnNamed: columnName [
12481255
"Answer the index of a column with a given name or signal an exception if the column with that name was not found"

0 commit comments

Comments
 (0)