test_plotter.py

from pathlib import Path

from Dataset import Dataset
from Plotter import Plotter
import pandas as pd

this_file_dir = Path(__file__).parent


def test_load_plotter():
    """Tests that the Plotter class can be loaded."""
    dataset = Dataset(this_file_dir / "../data/GamingStudy_data.csv")
    plotter = Plotter(dataset)
    assert type(plotter.df) == pd.DataFrame
    assert type(plotter.ds) == Dataset