DataFrame/src/DataFrame-Tests/DataFrameHeadTailTest.class.st at scatterplot · PolyMathOrg/DataFrame · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
Class {
	#name : #DataFrameHeadTailTest,
	#superclass : #TestCase,
	#instVars : [
		'df',
		'series'
	],
	#category : #'DataFrame-Tests-Core'
}

{ #category : #running }
DataFrameHeadTailTest >> setUp [

	super setUp.

	df := DataFrame withRows:
		      #( #( 5.1 3.5 1.4 0.2 setosa ) #( 4.9 3 1.4 0.2 setosa ) #( 4.7 3.2 1.3 0.2 setosa ) #( 7 3.2 4.7 1.4 versicolor ) #( 6.4 3.2 4.5 1.5 versicolor )
		         #( 6.9 3.1 4.9 1.5 versicolor ) #( 6.3 3.3 6 2.5 virginica ) #( 5.8 2.7 5.1 1.9 virginica ) #( 7.1 3 5.9 2.1 virginica ) ).

	df columnNames: #( sepalLength sepalWidth petalLength petalWidth species ).
	series := df column: #sepalLength
]

{ #category : #tests }
DataFrameHeadTailTest >> testDataFrameHead [

	| actual expected |

	expected := DataFrame withRows: #(
		(5.1 3.5 1.4 0.2 setosa)
		(4.9 3 1.4 0.2 setosa)
		(4.7 3.2 1.3 0.2 setosa)
		(7 3.2 4.7 1.4 versicolor)
		(6.4 3.2 4.5 1.5 versicolor)).
	expected columnNames: df columnNames.
	expected rowNames: (1 to: df defaultHeadTailSize).

	actual := df head.

	self assert: actual equals: expected
]

{ #category : #tests }
DataFrameHeadTailTest >> testDataFrameHeadN [

	| actual expected |

	expected := DataFrame withRows: #(
		(5.1 3.5 1.4 0.2 setosa)
		(4.9 3 1.4 0.2 setosa)
		(4.7 3.2 1.3 0.2 setosa)).
	expected columnNames: df columnNames.
	expected rowNames: (1 to: 3).

	actual := df head: 3.

	self assert: actual equals: expected
]

{ #category : #tests }
DataFrameHeadTailTest >> testDataFrameTail [

	| actual expected rows |

	expected := DataFrame withRows: #(
		(6.4 3.2 4.5 1.5 versicolor)
		(6.9 3.1 4.9 1.5 versicolor)
		(6.3 3.3 6 2.5 virginica)
		(5.8 2.7 5.1 1.9 virginica)
		(7.1 3 5.9 2.1 virginica)).
	expected columnNames: df columnNames.

	rows := df numberOfRows.
	expected rowNames:
		(rows - df defaultHeadTailSize + 1 to: rows).

	actual := df tail.

	self assert: actual equals: expected
]

{ #category : #tests }
DataFrameHeadTailTest >> testDataFrameTailN [

	| actual expected rows |

	expected := DataFrame withRows: #(
		(6.3 3.3 6 2.5 virginica)
		(5.8 2.7 5.1 1.9 virginica)
		(7.1 3 5.9 2.1 virginica)).
	expected columnNames: df columnNames.

	rows := df numberOfRows.
	expected rowNames:
		(rows - 3 + 1 to: rows).

	actual := df tail: 3.

	self assert: actual equals: expected
]

{ #category : #tests }
DataFrameHeadTailTest >> testDefaultHeadTailSize [

	self assert: df defaultHeadTailSize equals: 5
]