Change helper function to return single output dir

Author: dimkarakostas

tests/test_helper.py

@@ -68,10 +68,9 @@ def test_input_directories():
     assert len(input_dirs) > 0
 
 
-def test_output_directories():
-    output_dirs = hlp.get_output_directories()
-    assert isinstance(output_dirs, list)
-    assert len(output_dirs) > 0
+def test_output_directory():
+    output_dir = hlp.get_output_directory()
+    assert isinstance(output_dir, pathlib.Path)
 
 
 def test_tau_thresholds():
@@ -389,8 +388,8 @@ def test_get_output_row(mocker):
 
 
 def test_get_output_filename(mocker):
-    get_output_directories_mock = mocker.patch('tokenomics_decentralization.helper.get_output_directories')
-    get_output_directories_mock.return_value = [pathlib.Path(__file__).resolve().parent]
+    get_output_directory_mock = mocker.patch('tokenomics_decentralization.helper.get_output_directory')
+    get_output_directory_mock.return_value = pathlib.Path(__file__).resolve().parent
     get_exclude_contracts_mock = mocker.patch('tokenomics_decentralization.helper.get_exclude_contracts_flag')
     get_exclude_contracts_mock.return_value = False
     get_exclude_below_fees_mock = mocker.patch('tokenomics_decentralization.helper.get_exclude_below_fees_flag')

tokenomics_decentralization/helper.py

@@ -141,13 +141,13 @@ def increment_date(date, by):
         raise ValueError(f'Invalid granularity: {by}')
 
 
-def get_output_directories():
+def get_output_directory():
     """
     Reads the config file and retrieves the output directories
     :returns: a list of directories that might contain the db files
     """
     config = get_config_data()
-    return [pathlib.Path(db_dir).resolve() for db_dir in config['output_directories']]
+    return [pathlib.Path(db_dir).resolve() for db_dir in config['output_directories']][0]
 
 
 def get_input_directories():
@@ -496,7 +496,7 @@ def get_output_filename():
     if exclude_below_usd_cent_flag:
         output_filename += '-exclude_below_usd_cent'
     output_filename += '.csv'
-    return get_output_directories()[0] / output_filename
+    return get_output_directory() / output_filename
 
 
 def write_csv_output(output_rows):