Add test

PolyMathOrg · Aug 20, 2023 · 188c86c · 188c86c
1 parent 202adb8
commit 188c86c
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/src/DataFrame-IO-Tests/DataFrameCsvReaderTest.class.st b/src/DataFrame-IO-Tests/DataFrameCsvReaderTest.class.st
@@ -78,6 +78,28 @@ DataFrameCsvReaderTest >> testReadCsv [
 	self assert: actualDataFrame equals: dataFrameWithoutRowNames
 ]
 
+{ #category : #tests }
+DataFrameCsvReaderTest >> testReadCsvWithColumnNamesWithSeparator [
+
+	| controlDataFrame customColumnNames |
+
+	customColumnNames := #('Col1' 'Col2' 'Col3' 'Col4').
+
+	controlDataFrame := DataFrame 
+		readFromCsv: tabCsvFile 
+		withColumnNames: customColumnNames
+		separator: Character tab.
+
+	self 
+		assertCollection: controlDataFrame columnNames 
+		equals: customColumnNames asOrderedCollection.
+
+	"The original header is counted as a normal row now"
+	self 
+		assert: controlDataFrame size 
+		equals: 6.
+]
+
 { #category : #tests }
 DataFrameCsvReaderTest >> testReadCsvWithRowNames [
 	| actualDataFrame |