Wrote tests for queries

Author: olekscode

DataFrame-Core.package/DataFrame.class/instance/closeTo..st

@@ -0,0 +1,22 @@
+comparing
+closeTo: aDataFrame
+
+	aDataFrame species = self species
+		ifFalse: [ ^ false ].
+	
+	aDataFrame dimensions = self dimensions
+		ifFalse: [ ^ false ].
+		
+	aDataFrame columnTypes = self columnTypes
+		ifFalse: [ ^ false ].
+		
+	(aDataFrame rowNames = self rowNames
+		and: [ aDataFrame columnNames = self columnNames ])
+		ifFalse: [ ^ false ].
+		
+	1 to: self numberOfRows do: [ :i |
+		1 to: self numberOfColumns do: [ :j |
+			((self at: i at: j) closeTo: (aDataFrame at: i at: j))
+				ifFalse: [ ^ false ] ] ].
+	
+	^ true

DataFrame-Core.package/DataFrame.class/instance/select..st

@@ -16,7 +16,7 @@ select: aBlock
 		(selectedRowNumbers size @ self numberOfColumns).
 
 	df rowNames: (selectedRowNumbers collect: [ :i |
-		self rowNames at: i ]).
+		self rowNames at: i ]) asArray.
 
 	df columnNames: self columnNames.
 

DataFrame-Tests.package/DataFrameQueriesTests.class/README.md

@@ -0,0 +1,4 @@
+initialization
+setUp
+
+	df := DataFrame loadIris

DataFrame-Tests.package/DataFrameQueriesTests.class/instance/setUp.st

@@ -0,0 +1,10 @@
+initialization
+testSelectAllWhere
+
+	| actual |
+	
+	actual := df selectAllWhere: [ :species :sepal_width |
+		species = #setosa and: sepal_width = 3 ].
+	
+	self assert: actual columnNames equals: df columnNames.
+	self assert: actual rowNames equals: #(2 13 14 26 39 46).

DataFrame-Tests.package/DataFrameQueriesTests.class/instance/testSelectAllWhere.st

@@ -0,0 +1,14 @@
+initialization
+testSelectWhere
+
+	| selectedCols actual |
+	
+	selectedCols := #(petal_width petal_length).
+	
+	actual := df
+		select: selectedCols
+		where: [ :species :sepal_width |
+			species = #setosa and: sepal_width = 3 ].
+	
+	self assert: actual columnNames equals: selectedCols.
+	self assert: actual rowNames equals: #(2 13 14 26 39 46).

DataFrame-Tests.package/DataFrameQueriesTests.class/instance/testSelectWhere.st

@@ -0,0 +1,19 @@
+initialization
+testSelectWhereGroupbyAggregate
+
+	| selectedCols actual expected |
+	
+	selectedCols := #(petal_width petal_length).
+	
+	actual := df select: #(sepal_length species)
+		where: [ :petal_length :petal_width |
+			(petal_length < 4.9 and: petal_length > 1.6) and:
+			(petal_width < 0.4 or: petal_width > 1.5) ]
+		groupBy: #species
+		aggregate: #sum.
+		
+	expected := DataFrame fromColumns: #((15.9 18.2 17.1)).
+	expected columnNames: #(sepal_length).
+	expected rowNames: #(setosa versicolor virginica).
+
+	self assert: actual closeTo: expected.

DataFrame-Tests.package/DataFrameQueriesTests.class/instance/testSelectWhereGroupbyAggregate.st

@@ -0,0 +1,13 @@
+{
+	"commentStamp" : "",
+	"super" : "TestCase",
+	"category" : "DataFrame-Tests",
+	"classinstvars" : [ ],
+	"pools" : [ ],
+	"classvars" : [ ],
+	"instvars" : [
+		"df"
+	],
+	"name" : "DataFrameQueriesTests",
+	"type" : "normal"
+}

DataFrame-Tests.package/DataFrameQueriesTests.class/properties.json

@@ -20,8 +20,6 @@ Metacello new
   load.
 ```
 
-If you want to contribute to DataFrame library, you will need create a local DataFrame repository and add it to your Iceberg. This can be done automatically by the Metacello script provided above, but first you have to enable Metacello integration for Iceberg. To do that open the Settings Browser using _Ctrl+O+S_ or _World Menu -> System -> Settings_, then click on _Tools -> Software Configuration Management -> Iceberg_, and check the _Enable Metacello integration..._ checkbox. Now all you have to do is run a Metacello script - it will set up the DataFrame repository for you.
- 
 ### Creating DataSeries
 DataSeries can be created from an array of values