#numericalColumns now returns an array of all numerical columns

Author: Joshua-Dias-Barreto

src/DataFrame-Tests/DataFrameTest.class.st

@@ -2833,26 +2833,25 @@ DataFrameTest >> testNumericalColumnNames [
 		                #( Female 22 Spain 95 ) #( Male 24 Portugal 99 ) ).
 	dataFrame columnNames: #( Gender Age Country Score ).
 
-	expected := #( Age Score ) asOrderedCollection.
+	expected := #( Age Score ).
 
 	self assert: dataFrame numericalColumnNames equals: expected
 ]
 
 { #category : #tests }
 DataFrameTest >> testNumericalColumns [
 
-	| dataFrame expected |
+	| dataFrame expectedCollection |
 	dataFrame := DataFrame withRows:
 		             #( #( Male 21 Argentina 94 ) #( Female 20 France 97 )
 		                #( Female 22 Spain 95 ) #( Male 24 Portugal 99 ) ).
 	dataFrame columnNames: #( Gender Age Country Score ).
 
-	expected := DataFrame withRows:
-		            #( #( 21 94 ) #( 20 97 ) #( 22 95 ) #( 24 99 ) ).
-
-	expected columnNames: #( Age Score ).
+	expectedCollection := {
+		                      (dataFrame columnAt: 2).
+		                      (dataFrame columnAt: 4) } asArray.
 
-	self assert: dataFrame numericalColumns equals: expected
+	self assert: dataFrame numericalColumns equals: expectedCollection
 ]
 
 { #category : #splitjoin }

src/DataFrame/DataFrame.class.st

@@ -1431,15 +1431,15 @@ DataFrame >> numberOfRows [
 DataFrame >> numericalColumnNames [
 	"Returns the names of all numerical columns of the dataframe"
 
-	^ self columnNames select: [ :columnName |
-		  (self dataTypes at: columnName) includesBehavior: Number ]
+	^ self numericalColumns collect: [ :column | column name ]
 ]
 
 { #category : #accessing }
 DataFrame >> numericalColumns [
 	"Returns all numerical columns of the dataframe"
 
-	^ self columns: self numericalColumnNames
+	^ self columns select: [ :column |
+		  (self dataTypes at: column name) includesBehavior: Number ]
 ]
 
 { #category : #splitjoin }