Implemented add:atKey: message for DataSeries

Author: olekscode

DataFrame-Core.package/DataSeries.class/instance/add.atKey..st

@@ -0,0 +1,6 @@
+adding
+add: aValue atKey: aKey
+
+	contents add: aValue.
+	keys := keys copyWith: aKey.
+	self induceTypeWithNewElement: aValue.

DataFrame-Core.package/DataSeries.class/instance/at.put..st

@@ -1,16 +1,5 @@
 accessing
-at: aNumber put: value
+at: aNumber put: aValue
 
-	contents at: aNumber put: value.
-	
-	self type isNil
-		"In case we fill the empty series with elements (e.g. collect)"
-		ifTrue: [ type := value class ]
-		
-		"In case we add an element to the existing series"
-		ifFalse: [ 
-			| inductor |
-			inductor := DataTypeInductor new.
-			type := inductor
-				leastCommonSuperclassOf: value class
-				and: self type. ].
+	contents at: aNumber put: aValue.
+	self induceTypeWithNewElement: aValue.

DataFrame-Core.package/DataSeries.class/instance/atKey.put..st

@@ -7,8 +7,7 @@ atKey: aKey put: aValue
 	index = 0
 		"a key was not found - create a new element"
 		ifTrue: [
-			keys := keys copyWith: aKey.
-			contents add: aValue ]
+			self add: aValue atKey: aKey ]
 		"a key was found - update the value"
 		ifFalse: [
 			self at: index put: aValue ].

DataFrame-Core.package/DataSeries.class/instance/induceTypeWithNewElement..st

@@ -0,0 +1,13 @@
+as yet unclassified
+induceTypeWithNewElement: aValue
+"Induces type of series given that the new element was added to it"
+
+	self type isNil
+		"In case we fill the empty series with elements (e.g. collect)"
+		ifTrue: [ type := aValue class ]
+		
+		"In case we add an element to the existing series"
+		ifFalse: [ 
+			type := DataTypeInductor new
+				leastCommonSuperclassOf: aValue class
+				and: self type. ].

DataFrame-Tests.package/DataSeriesTests.class/instance/testAddAtKey.st

@@ -0,0 +1,11 @@
+tests
+testAddAtKey
+
+	| expected |
+	
+	series add: -2 atKey: #X.	
+	expected := #(10 20 30 40 50 60 70 80 90 100 -2) asDataSeries.
+	expected keys: ((1 to: 10) copyWith: #X).
+	expected name: series name.
+	
+	self assert: series equals: expected.

DataFrame-Tests.package/DataSeriesTests.class/instance/testAtKeyPut.st

@@ -0,0 +1,10 @@
+tests
+testAtKeyPut
+
+	| expected |
+	
+	series atKey: 6 put: -2.	
+	expected := #(10 20 30 40 50 -2 70 80 90 100) asDataSeries.
+	expected name: series name.
+	
+	self assert: series equals: expected.

DataFrame-Tests.package/DataSeriesTests.class/instance/testAtKeyPutNewElement.st

@@ -0,0 +1,11 @@
+tests
+testAtKeyPutNewElement
+
+	| expected |
+	
+	series atKey: #X put: -2.	
+	expected := #(10 20 30 40 50 60 70 80 90 100 -2) asDataSeries.
+	expected keys: ((1 to: 10) copyWith: #X).
+	expected name: series name.
+	
+	self assert: series equals: expected.