diff --git a/tests/test_hdtimeline.py b/tests/test_hdtimeline.py index 736cabb..bd9cc34 100644 --- a/tests/test_hdtimeline.py +++ b/tests/test_hdtimeline.py @@ -1,11 +1,19 @@ import sys +import glob sys.path.insert(0,".") # For Github +sys.path.insert(0,"./hdtimelines") # in case this is run when a submodule from hdtimelines import hdtimeline def test_hdtimeline(): + # Find test data path + if glob.glob('./hdtimelines/test_data/'): + path = './hdtimelines/test_data' + else: + path = './test_data' + hd = hdtimeline.hdTimeLine("Test timeline") - hd.add_topic_csv('Monarchs extract','./test_data/British Monarchs_extract_ok.csv') + hd.add_topic_csv('Monarchs extract',f'{path}/British Monarchs_extract_ok.csv') daterange = hd.get_date_range() assert hd.title == "Test timeline" @@ -21,7 +29,7 @@ def test_hdtimeline(): 'url': 'https://en.wikipedia.org/wiki/William_the_Conqueror'} assert hd.topics[0].events[0] == dtest - hd.add_topic_csv('Playwrights extract','./test_data/Playwrights_extract_ok.csv') + hd.add_topic_csv('Playwrights extract',f'{path}/Playwrights_extract_ok.csv') hd.move_topic(id=1, indexshift=1) # Move Monarchs down assert [topic.id for topic in hd.topics] == [2, 1] diff --git a/tests/test_pltimeline.py b/tests/test_pltimeline.py index 6286582..42f7146 100644 --- a/tests/test_pltimeline.py +++ b/tests/test_pltimeline.py @@ -1,5 +1,6 @@ import sys sys.path.insert(0,".") # For Github +sys.path.insert(0,"./hdtimelines") # in case this is run when a submodule from hdtimelines import pltimeline from historicaldate import hdateutils diff --git a/tests/test_pltutils.py b/tests/test_pltutils.py index 4a4284d..f0ec90a 100644 --- a/tests/test_pltutils.py +++ b/tests/test_pltutils.py @@ -1,20 +1,28 @@ import sys sys.path.insert(0,".") # For Github +sys.path.insert(0,"./hdtimelines") # in case this is run when a submodule +import glob from hdtimelines import pltutils import pandas as pd def test_check_dataframe(): - df = pd.read_csv('./test_data/Playwrights_extract_ok.csv', na_filter=False) + # Find test data path + if glob.glob('./hdtimelines/test_data/'): + path = './hdtimelines/test_data' + else: + path = './test_data' + + df = pd.read_csv(f'{path}/Playwrights_extract_ok.csv', na_filter=False) result, message = pltutils.check_dataframe(df) assert result - df = pd.read_csv('./test_data/British Monarchs_extract_ok.csv', na_filter=False) + df = pd.read_csv(f'{path}/British Monarchs_extract_ok.csv', na_filter=False) result, message = pltutils.check_dataframe(df) assert result assert pltutils.check_dataframe(df) - df = pd.read_csv('./test_data/British Monarchs_extract_notok_no_label.csv', na_filter=False) + df = pd.read_csv(f'{path}/British Monarchs_extract_notok_no_label.csv', na_filter=False) result, message = pltutils.check_dataframe(df) assert not result assert message == "Error: KeyError('label')"