fix tests

graphsense · Feb 28, 2024 · 2548583 · 2548583
1 parent e09e54c
commit 2548583
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/src/test/scala/org/graphsense/account/eth/BalancesTest.scala b/src/test/scala/org/graphsense/account/eth/BalancesTest.scala
@@ -10,7 +10,7 @@ class BalancesTest extends TestBase {
   private val refDir = inputDir + "reference/"
 
   private val ds = new TestEthSource(spark, inputDir)
-  private val t = new EthTransformation(spark, 2, 100000)
+  private val t = new EthTransformation(spark, 2, 100000, 100)
 
   import spark.implicits._
 

diff --git a/src/test/scala/org/graphsense/account/eth/ComplexGraphTransformationTest.scala b/src/test/scala/org/graphsense/account/eth/ComplexGraphTransformationTest.scala
@@ -19,7 +19,7 @@ class ComplexGraphTransformationTest extends TestBase {
   private val inputDir = "src/test/resources/account/eth/complex_graph/"
   private val refDir = inputDir + "reference/"
   private val ds = new TestEthSource(spark, inputDir)
-  private val t = new EthTransformation(spark, 2, 100000)
+  private val t = new EthTransformation(spark, 2, 100000, 100)
 
   // read raw data
   val blocks = ds.blocks()
@@ -75,7 +75,7 @@ class ComplexGraphTransformationTest extends TestBase {
   addressRelations.write
     .mode("overwrite")
     .json("test_ref/address_relations.json")
-  
+
   // test equality
 
   note("Testing address graph:")

diff --git a/src/test/scala/org/graphsense/account/eth/TokensTest.scala b/src/test/scala/org/graphsense/account/eth/TokensTest.scala
@@ -13,7 +13,7 @@ class TokenTest extends TestBase {
   private val inputDir = "src/test/resources/account/eth/tokens/"
 
   private val ds = new TestEthSource(spark, inputDir)
-  private val t = new EthTransformation(spark, 2, 100000)
+  private val t = new EthTransformation(spark, 2, 100000, 100)
 
   test("full transform with logs") {
     // Parse all but only keep USDT token transfers for compare but parse all of them