def test_get_collections_type(self):
df = get_collections("Industrial Services",
"sector",
output_format='pandas')
assert isinstance(df, pd.DataFrame)
assert len(df) == 39
assert len(df.columns) > 500
def test_get_collections_pandas(self):
df = get_collections("Restaurants", output_format='pandas')
assert isinstance(df, pd.DataFrame)
assert "change" in df.index
assert "close" in df.index
def test_get_collections_pandas(self):
df = get_collections("Computer Hardware", output_format='pandas')
assert isinstance(df, pd.DataFrame)
assert len(df) == 39
assert "change" in df.index
assert "close" in df.index
def test_get_collections(self):
# by tag
data = get_collections("Restaurants")
assert isinstance(data, list)
assert len(data) > 100
def test_get_collections_no_collection(self):
with pytest.raises(TypeError):
get_collections()
with pytest.raises(ValueError):
get_collections("Computer Hardware", "badcollection")
def test_collections(self):
data = get_collections("Technology", "sector")
assert isinstance(data, pd.DataFrame)
assert "companyName" in data.columns
def test_collections_empty_name(self):
data = get_collections("BADNAME", "sector")
assert data.empty
def test_collections_bad_type(self):
with pytest.raises(TypeError):
get_collections()
with pytest.raises(ValueError):
get_collections("Technology", "BADTYPE")
def test_get_collections(self):
data = get_collections("Computer Hardware")
assert isinstance(data, list)
assert len(data) > 100
