- Added test for new sheet implementation

Signed-off-by: LeandroC89 <[email protected]>

Author: LeandroC89

.gitignore

@@ -3,3 +3,4 @@
 .gradle
 build
 .ipynb_checkpoints
+/dataframe-excel/src/test/resources/generated_wb.xlsx

dataframe-excel/src/test/kotlin/org/jetbrains/kotlinx/dataframe/io/XlsxTest.kt

@@ -109,4 +109,18 @@ class XlsxTest {
             DataFrame.readExcel(testResource("xlsx6.xlsx"), skipRows = 4)
         }
     }
+
+    @Test
+    fun `write to new sheet when erase is false`() {
+        val df = DataFrame.readExcel(testResource("sample4.xls"), skipRows = 6, rowsCount = 1)
+        val fileLoc = testResource("generated_wb.xlsx").toURI().toString().removeRange(0, 6)
+println(fileLoc)
+        df.writeExcel(fileLoc, sheetName = "TestSheet1")
+        df.writeExcel(fileLoc, sheetName = "TestSheet2")
+
+        val testSheet1Df = DataFrame.readExcel(testResource("generated_wb.xlsx"), sheetName = "TestSheet1")
+        val testSheet2Df = DataFrame.readExcel(testResource("generated_wb.xlsx"), sheetName = "TestSheet2")
+
+        testSheet1Df.columnNames() shouldBe testSheet2Df.columnNames()
+    }
 }