hazza/docker-stacks
1
0
Fork 0
mirror of https://github.com/jupyter/docker-stacks.git synced 2025-10-08 10:34:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Improve some tests (#2278)

Browse Source 
* Improve some tests

* Fix
This commit is contained in:
Ayaz Salikhov
2025-04-04 01:44:03 +01:00
committed by GitHub
parent f4b31dda77
commit c81d61bbc1
3 changed files with 6 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/by_image/all-spark-notebook/test_spark_r_nbconvert.py
View File

@@ -35,7 +35,5 @@ def test_spark_r_nbconvert(
no_warnings=(not expected_warnings), no_warnings=(not expected_warnings),
) )
if expected_warnings: warnings = TrackedContainer.get_warnings(logs)
warnings = TrackedContainer.get_warnings(logs) assert warnings == expected_warnings
assert len(warnings) == len(expected_warnings)
assert expected_warnings[0] == warnings[0]

4
tests/by_image/pyspark-notebook/test_spark.py
View File

@@ -15,7 +15,5 @@ def test_spark_shell(container: TrackedContainer) -> None:
command=["bash", "-c", 'spark-shell <<< "1+1"'], command=["bash", "-c", 'spark-shell <<< "1+1"'],
) )
warnings = TrackedContainer.get_warnings(logs) warnings = TrackedContainer.get_warnings(logs)
assert len(warnings) == 1 assert warnings == ["WARNING: Using incubator modules: jdk.incubator.vector"]
assert "Using incubator modules: jdk.incubator.vector" in warnings[0]
assert "res0: Int = 2" in logs, "spark-shell does not work" assert "res0: Int = 2" in logs, "spark-shell does not work"

10
tests/by_image/pyspark-notebook/test_spark_nbconvert.py
View File

@@ -12,19 +12,15 @@ LOGGER = logging.getLogger(__name__)
THIS_DIR = Path(__file__).parent.resolve() THIS_DIR = Path(__file__).parent.resolve()
@pytest.mark.parametrize( @pytest.mark.parametrize("test_file", ["issue_1168", "local_pyspark"])
"test_file",
["issue_1168", "local_pyspark"],
)
@pytest.mark.parametrize("output_format", ["pdf", "html", "markdown"]) @pytest.mark.parametrize("output_format", ["pdf", "html", "markdown"])
def test_spark_nbconvert( def test_spark_nbconvert(
container: TrackedContainer, test_file: str, output_format: str container: TrackedContainer, test_file: str, output_format: str
) -> None: ) -> None:
host_data_file = THIS_DIR / "data" / f"{test_file}.ipynb" host_data_file = THIS_DIR / "data" / f"{test_file}.ipynb"
logs = check_nbconvert( logs = check_nbconvert(
container, host_data_file, "markdown", execute=True, no_warnings=False container, host_data_file, output_format, execute=True, no_warnings=False
) )
warnings = TrackedContainer.get_warnings(logs) warnings = TrackedContainer.get_warnings(logs)
assert len(warnings) == 1 assert warnings == ["WARNING: Using incubator modules: jdk.incubator.vector"]
assert "Using incubator modules: jdk.incubator.vector" in warnings[0]