Skip to content
Snippets Groups Projects
Commit abde47b8 authored by lukas leufen's avatar lukas leufen
Browse files

added test for helper function to_list

parent 20d2d659
Branches
Tags
2 merge requests!9new version v0.2.0,!8data generator
...@@ -8,6 +8,7 @@ import logging ...@@ -8,6 +8,7 @@ import logging
import pandas as pd import pandas as pd
import datetime as dt import datetime as dt
from typing import Iterator, Union, List from typing import Iterator, Union, List
from src import helpers
join_url_base = 'https://join.fz-juelich.de/services/rest/surfacedata/' join_url_base = 'https://join.fz-juelich.de/services/rest/surfacedata/'
logging.basicConfig(level=logging.INFO) logging.basicConfig(level=logging.INFO)
...@@ -24,8 +25,7 @@ def download_join(station_name: Union[str, List[str]], statvar: dict) -> [pd.Dat ...@@ -24,8 +25,7 @@ def download_join(station_name: Union[str, List[str]], statvar: dict) -> [pd.Dat
- meta - pandas df with all meta information - meta - pandas df with all meta information
""" """
# make sure station_name parameter is a list # make sure station_name parameter is a list
if not isinstance(station_name, list): station_name = helpers.to_list(station_name)
station_name = [station_name]
# load series information # load series information
opts = {'base': join_url_base, 'service': 'series', 'station_id': station_name[0]} opts = {'base': join_url_base, 'service': 'series', 'station_id': station_name[0]}
......
import pytest
from src.helpers import to_list
class TestToList:
def test_to_list(self):
assert to_list('a') == ['a']
assert to_list('abcd') == ['abcd']
assert to_list([1, 2, 3]) == [1, 2, 3]
assert to_list([45]) == [45]
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment