Skip to content

Commit bc6cd84

Browse files
Added a test for decimal and large values.
1 parent 562c281 commit bc6cd84

1 file changed

Lines changed: 28 additions & 12 deletions

File tree

src/DataFrame-Tests/DataSeriesTest.class.st

Lines changed: 28 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -1279,10 +1279,10 @@ DataSeriesTest >> testEncodeOneHot [
12791279
| actual expected |
12801280
actual := #( 1 2 3 4 ) asDataSeries encodeOneHot.
12811281
expected := #(
1282-
#(1 0 0 0)
1283-
#(0 1 0 0)
1284-
#(0 0 1 0)
1285-
#(0 0 0 1)
1282+
#( 1 0 0 0 )
1283+
#( 0 1 0 0 )
1284+
#( 0 0 1 0 )
1285+
#( 0 0 0 1 )
12861286
) asDataSeries.
12871287

12881288
self assert: actual equals: expected
@@ -1295,10 +1295,10 @@ DataSeriesTest >> testEncodeOneHotRomanNumbers [
12951295
actual := (#( I XIV VII XXXII ) collect: [ :each | each romanNumber ])
12961296
asDataSeries encodeOneHot.
12971297
expected := #(
1298-
#(1 0 0 0)
1299-
#(0 0 1 0)
1300-
#(0 1 0 0)
1301-
#(0 0 0 1)
1298+
#( 1 0 0 0 )
1299+
#( 0 0 1 0 )
1300+
#( 0 1 0 0 )
1301+
#( 0 0 0 1 )
13021302
) asDataSeries.
13031303

13041304
self assert: actual equals: expected
@@ -1310,10 +1310,26 @@ DataSeriesTest >> testEncodeOneHotStrings [
13101310
| actual expected |
13111311
actual := #( apple avocado orange banana ) asDataSeries encodeOneHot.
13121312
expected := #(
1313-
#(1 0 0 0)
1314-
#(0 1 0 0)
1315-
#(0 0 0 1)
1316-
#(0 0 1 0)
1313+
#( 1 0 0 0 )
1314+
#( 0 1 0 0 )
1315+
#( 0 0 0 1 )
1316+
#( 0 0 1 0 )
1317+
) asDataSeries.
1318+
1319+
self assert: actual equals: expected
1320+
]
1321+
1322+
{ #category : #'tests - converting' }
1323+
DataSeriesTest >> testEncodeOneHotWithDecimalAndLargeValues [
1324+
1325+
| actual expected |
1326+
actual := #( 0.5 1 2.2 3 4000 ) asDataSeries encodeOneHot.
1327+
expected := #(
1328+
#( 1 0 0 0 0 )
1329+
#( 0 1 0 0 0 )
1330+
#( 0 0 1 0 0 )
1331+
#( 0 0 0 1 0 )
1332+
#( 0 0 0 0 1 )
13171333
) asDataSeries.
13181334

13191335
self assert: actual equals: expected

0 commit comments

Comments
 (0)