Merge pull request #264 from Joshua-Dias-Barreto/chainedSorting

Fixes #263. Added Collection>> #asDataFrame with tests.

Author: jecisc

src/DataFrame-Tests/DataFrameTest.class.st

@@ -421,6 +421,52 @@ DataFrameTest >> testAsArrayOfRowsWithName [
 		equals: expected asOrderedCollection
 ]
 
+{ #category : #tests }
+DataFrameTest >> testAsDataFrame [
+
+	| actual expected |
+	actual := #( #( 1 2 3 ) 
+					#( 4 5 6 ) 
+					#( 7 8 9 ) ) asDataFrame.
+	expected := DataFrame withRows: #( #( 1 2 3 ) #( 4 5 6 ) #( 7 8 9 ) ).
+
+	self assert: actual equals: expected
+]
+
+{ #category : #tests }
+DataFrameTest >> testAsDataFrameEmpty [
+
+	| actual expected |
+	actual := #() asDataFrame.
+	expected := DataFrame new.
+
+	self assert: actual equals: expected
+]
+
+{ #category : #tests }
+DataFrameTest >> testAsDataFrameEmptyColumns [
+
+	| actual expected |
+	actual := #( #() 
+					 #() 
+					 #() ) asDataFrame.
+	expected := DataFrame new: 3 @ 0.
+
+	self assert: actual equals: expected
+]
+
+{ #category : #tests }
+DataFrameTest >> testAsDataFrameWithUnevenRowElements [
+
+	| actual expected |
+	actual := #( #( 1 2 ) 
+					#( 3 4 5 ) 
+					#( 6 7 8 9  ) ) asDataFrame.
+	expected := DataFrame withRows: #( #( 1 2 ) #( 3 4 5 ) #( 6 7 8 9 ) ).
+
+	self assert: actual equals: expected
+]
+
 { #category : #tests }
 DataFrameTest >> testAt [
 

src/DataFrame/Collection.extension.st

@@ -6,6 +6,28 @@ Collection >> ** arg [
 	^ self raisedTo: arg
 ]
 
+{ #category : #'*DataFrame' }
+Collection >> asDataFrame [
+
+	| numberOfRows numberOfColumns dataFrame |
+	numberOfRows := self size.
+	numberOfColumns := 0.
+	numberOfRows = 0 ifFalse: [
+		numberOfColumns := (self collect: #size) max ].
+	dataFrame := DataFrame new: numberOfRows @ numberOfColumns.
+
+	1 to: numberOfRows do: [ :rowIndex |
+		| row |
+		row := self at: rowIndex.
+		1 to: numberOfColumns do: [ :colIndex |
+			| value |
+			value := row at: colIndex ifAbsent: [ nil ].
+			dataFrame at: rowIndex at: colIndex put: value ] ].
+
+
+	^ dataFrame
+]
+
 { #category : #'*DataFrame-Core-Base' }
 Collection >> asDataSeries [