def test_get_first_page_of_companies_data():
companies = get_page('companies')
assert isinstance(companies, pd.DataFrame)
assert len(companies) == get_page_size('companies')
assert companies.total_pages > 1
assert 'ticker' in companies.columns
assert 'cik' in companies.columns
assert 'lei' in companies.columns
assert 'name' in companies.columns
def test_get_page_size_with_undefined_endpoint():
page_size = get_page_size('not-defined')
assert page_size == 250
def test_get_page_size_with_defined_endpoint():
page_size = get_page_size('prices')
assert page_size == 50000
def test_get_companies_data_with_query_that_returns_many_pages():
companies = get('companies', query='Corp')
assert len(companies) > get_page_size('companies')
assert len(companies) > 300
assert companies.name.str.contains('Corp').any()