diff --git a/requirements.txt b/requirements.txt index aa951aa..610d6ad 100644 --- a/requirements.txt +++ b/requirements.txt @@ -4,3 +4,5 @@ requests lxml tqdm python-decouple + +tox diff --git a/test_api.py b/test_api.py new file mode 100755 index 0000000..ff97558 --- /dev/null +++ b/test_api.py @@ -0,0 +1,38 @@ +from auto_scheduler.api import get_transmitter_stats, \ + get_active_transmitter_info + +def test_get_transmitter_stats(): + transmitters = get_transmitter_stats() + + for t in transmitters: + if t['uuid'] == "qW5N27QuSrN2JMasbNiUhR": + print(t) + + assert(type(transmitters) == list) + transmitters_keys = transmitters[0].keys() + for key in ['uuid', + 'description', + 'alive', + 'type', + 'mode', + 'satellite', + 'norad_cat_id', + 'success_rate', + 'bad_rate', + 'unvetted_rate', + 'good_count', + 'bad_count', + 'unvetted_count', + 'data_count']: + assert(key in transmitters_keys) + + +def test_get_active_transmitter_info(): + transmitters = get_active_transmitter_info(135000000, + 148000000) + + assert(type(transmitters) == list) + transmitters_keys = transmitters[0].keys() + for key in ['uuid', + 'norad_cat_id']: + assert(key in transmitters_keys)