53 lines
1.5 KiB
Python
53 lines
1.5 KiB
Python
"""Tests of the creation and the export of the scores table."""
|
|
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
pandas = pytest.importorskip("pandas")
|
|
|
|
from weather import export # noqa: E402
|
|
|
|
|
|
def test_create_scores_frame_has_the_columns_and_values_of_the_lecture() -> None:
|
|
frame = export.create_scores_frame()
|
|
|
|
assert isinstance(frame, pandas.DataFrame)
|
|
assert list(frame.columns) == ["students", "scores"]
|
|
assert frame["students"].to_list() == ["Amy", "James", "Angela"]
|
|
assert frame["scores"].to_list() == [76, 56, 65]
|
|
|
|
|
|
def test_export_frame_round_trips_through_a_file(tmp_path: Path) -> None:
|
|
frame = export.create_scores_frame()
|
|
target = tmp_path / "scores.csv"
|
|
|
|
export.export_frame(frame, target)
|
|
|
|
assert target.is_file()
|
|
read_back = pandas.read_csv(target, index_col=0)
|
|
pandas.testing.assert_frame_equal(read_back, frame)
|
|
|
|
|
|
def test_export_frame_accepts_a_string_path(tmp_path: Path) -> None:
|
|
target = tmp_path / "as_text.csv"
|
|
|
|
export.export_frame(export.create_scores_frame(), str(target))
|
|
|
|
assert target.is_file()
|
|
|
|
|
|
def test_export_frame_default_name_writes_only_new_data_csv_in_the_working_dir(
|
|
tmp_path: Path,
|
|
) -> None:
|
|
# The autouse fixture in conftest.py made tmp_path the working directory.
|
|
export.export_frame(export.create_scores_frame())
|
|
|
|
assert [item.name for item in tmp_path.iterdir()] == ["new_data.csv"]
|
|
|
|
|
|
def test_create_scores_frame_writes_nothing(tmp_path: Path) -> None:
|
|
export.create_scores_frame()
|
|
|
|
assert list(tmp_path.iterdir()) == []
|