fixed numberOfShards in single server tests

Author: rashtao

integration-tests/src/test/scala/org/apache/spark/sql/arangodb/datasource/BaseSparkTest.scala

@@ -34,6 +34,7 @@ class BaseSparkTest {
   }
 
   def isSingle: Boolean = BaseSparkTest.isSingle
+  def isCluster: Boolean = !BaseSparkTest.isSingle
 }
 
 object BaseSparkTest {

integration-tests/src/test/scala/org/apache/spark/sql/arangodb/datasource/write/CreateCollectionTest.scala

@@ -50,7 +50,9 @@ class CreateCollectionTest extends BaseSparkTest {
       ))
       .save()
 
-    assertThat(collection.getProperties.getNumberOfShards).isEqualTo(5)
+    if (isCluster) {
+      assertThat(collection.getProperties.getNumberOfShards).isEqualTo(5)
+    }
     assertThat(collection.getProperties.getType.getType).isEqualTo(CollectionType.EDGES.getType)
   }
 
@@ -70,7 +72,9 @@ class CreateCollectionTest extends BaseSparkTest {
       ))
       .save()
 
-    assertThat(collection.getProperties.getNumberOfShards).isEqualTo(5)
+    if (isCluster) {
+      assertThat(collection.getProperties.getNumberOfShards).isEqualTo(5)
+    }
     assertThat(collection.getProperties.getType.getType).isEqualTo(CollectionType.EDGES.getType)
   }