Skip to content

Commit

Permalink
Update tests scripts for data module
Browse files Browse the repository at this point in the history
  • Loading branch information
rosericazondekon committed Jun 6, 2023
1 parent 5af5061 commit 9d4310d
Showing 1 changed file with 12 additions and 1 deletion.
13 changes: 12 additions & 1 deletion tests/test_data.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
import pytest
import pandas as pd
from pynssp import load_simulated_ts, get_scenario1, get_scenario2
from pynssp import load_simulated_ts, get_scenario1, get_scenario2, load_nssp_stopwords


@pytest.fixture
Expand All @@ -18,6 +18,11 @@ def scenario2():
return get_scenario2()


@pytest.fixture
def nssp_stopwords():
return load_nssp_stopwords()


def test_load_simulated_ts(simulated_ts):
assert isinstance(simulated_ts, pd.DataFrame)
assert len(simulated_ts) == 626
Expand All @@ -34,3 +39,9 @@ def test_get_scenario2(scenario2):
assert isinstance(scenario2, pd.DataFrame)
assert len(scenario2) == 313
assert len(scenario2.columns) == 6


def test_load_nssp_stopwords(nssp_stopwords):
assert isinstance(nssp_stopwords, pd.DataFrame)
assert len(nssp_stopwords) == 835
assert len(nssp_stopwords.columns) == 2

0 comments on commit 9d4310d

Please sign in to comment.