Skip to content

Commit

Permalink
fix streaming test
Browse files Browse the repository at this point in the history
  • Loading branch information
SajidAlamQB committed Aug 16, 2023
1 parent a6867c7 commit 3b85fe1
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions kedro-datasets/tests/spark/test_spark_streaming_dataset.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,9 +4,11 @@
import pytest
from kedro.io.core import DataSetError
from moto import mock_s3
from pyspark import __version__
from pyspark.sql import SparkSession
from pyspark.sql.types import IntegerType, StringType, StructField, StructType
from pyspark.sql.utils import AnalysisException
from semver import VersionInfo

from kedro_datasets.spark.spark_dataset import SparkDataSet
from kedro_datasets.spark.spark_streaming_dataset import SparkStreamingDataSet
Expand All @@ -15,6 +17,7 @@
BUCKET_NAME = "test_bucket"
AWS_CREDENTIALS = {"key": "FAKE_ACCESS_KEY", "secret": "FAKE_SECRET_KEY"}

SPARK_VERSION = VersionInfo.parse(__version__)

def sample_schema(schema_path):
"""read the schema file from json path"""
Expand Down

0 comments on commit 3b85fe1

Please sign in to comment.