iSharkFly-Docs/java-tutorials
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

JAVA-12627 Fix DatasetUnitTest in apache-spark module

Browse Source
This commit is contained in:
anuragkumawat 2022-07-10 13:40:05 +05:30
parent a07710d011
commit eda84c24b9
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
apache-spark/src/test/java/com/baeldung/dataframes/CustomerToDataFrameConverterAppUnitTest.java
View File

@ -8,6 +8,8 @@ import java.util.List;
import org.apache.spark.sql.Dataset; import org.apache.spark.sql.Dataset;
import org.apache.spark.sql.Row; import org.apache.spark.sql.Row;
import org.apache.spark.sql.SparkSession;
import org.junit.jupiter.api.AfterEach;
import org.junit.jupiter.api.Test; import org.junit.jupiter.api.Test;
class CustomerToDataFrameConverterAppUnitTest { class CustomerToDataFrameConverterAppUnitTest {
@ -59,4 +61,9 @@ class CustomerToDataFrameConverterAppUnitTest {
assertEquals( "Male", row2.getAs("gender")); assertEquals( "Male", row2.getAs("gender"));
assertEquals( 1200, (int)row2.getAs("transaction_amount")); assertEquals( 1200, (int)row2.getAs("transaction_amount"));
} }
@AfterEach
public void cleanup() {
SparkSession.getActiveSession().get().close();
}
} }