Coverage for /home/runner/work/pycax/pycax/pycax/datasets/test_datasets.py: 100%

1"""Tests for datasets module""" 

2import requests 

3 

4from pycax import datasets 

5 

6from pycax.caxutils import cax_api_key 

7 

8 

9def test_datasets_get_data(): 

10 "datasets.get - test data" 

11 query = datasets.get() 

12 assert "DatasetsResponse" == query.__class__.__name__ 

13 assert not query.data 

14 query.execute() 

15 assert query.data.__class__.__name__ == "dict" 

16 assert 1 == len(query.data) 

17 assert dict == query.data["tables"][0].__class__ 

18 assert query.to_pandas().__class__.__name__ == "DataFrame" 

19 

20def test_datasets_get_url(): 

21 "datasets.get - test api_url" 

22 query = datasets.get() 

23 assert requests.get(query.api_url, params = {'XApiKey': cax_api_key}).status_code == 200 

24 

25def test_datasets_getdf_result(): 

26 "datasets.getdf - test result" 

27 res = datasets.getdf() 

28 assert res.__class__.__name__ == "DataFrame" 

29