def setUp(self):
# simulate getting a JSON response from a search query
mock_json(path="seriesbr.ibge.get_json",
json=get_sample_json("ibge_metadata.json")).start()
# simulate getting a JSON response from a metadata query
mock_json(
path="seriesbr.helpers.metadata.get_json",
json=get_sample_json("ibge_metadata.json"),
).start()
def test_bcb_parser(self):
df = bcb_json_to_df(get_sample_json("bcb_json.json"), 11, "Selic")
# assert it is a DataFrame
self.assertIsInstance(df, pd.DataFrame)
# index is datetime
self.assertTrue(pd.api.types.is_datetime64_dtype(df.index))
# series is float
self.assertTrue(pd.api.types.is_float_dtype(df.values))
def test_ipea_parser(self):
df = ipea_json_to_df(get_sample_json("ipea_json.json"), "Código",
"Nome")
# assert it is a DataFrame
self.assertIsInstance(df, pd.DataFrame)
# index is datetime
self.assertTrue(pd.api.types.is_datetime64_dtype(df.index))
# series is float
self.assertTrue(pd.api.types.is_float_dtype(df.values))
def test_bcb_get_search_results(self, mocked_get_json):
mocked_get_json.return_value = get_sample_json("bcb_search_results.json")
df = bcb_get_search_results("https://fake.com?json=call")
test = df.columns.tolist()
expected = ["codigo_sgs", "title", "periodicidade", "unidade_medida"]
self.assertListEqual(test, expected)
def setUp(self):
mock_json(path="seriesbr.ibge.get_json",
json=get_sample_json("ibge_search_results.json")).start()
def test_ibge_get_metadata(self, mocked_get_json):
mocked_get_json.return_value = get_sample_json("ibge_metadata.json")
df = ibge.get_metadata(1419)
self.assertFalse(df.empty)
def test_list_cities(self, mocked_get_json):
mocked_get_json.return_value = get_sample_json("ibge_municipios.json")
df = ibge.list_cities()
self.assertFalse(df.empty)
def test_list_mesoregions(self, mocked_get_json):
mocked_get_json.return_value = get_sample_json(
"ibge_mesorregioes.json")
df = ibge.list_mesoregions()
self.assertFalse(df.empty)
def setUp(self):
mock_json(path="seriesbr.ibge.get_json",
json=get_sample_json("ibge_variables.json")).start()
def setUp(self):
mock_json(
path="seriesbr.helpers.metadata.get_json",
json=get_sample_json("bcb_metadata.json"),
).start()
def test_list_countries(self, mocked_get_json):
mocked_get_json.return_value = get_sample_json("ipea_paises.json")
test = ipea.list_countries().columns.tolist()
expected = ["PAICODIGO", "PAINOME"]
self.assertListEqual(test, expected)
def test_list_themes(self, mocked_get_json):
mocked_get_json.return_value = get_sample_json("ipea_temas.json")
test = ipea.list_themes().columns.tolist()
expected = ["TEMCODIGO", "TEMCODIGO_PAI", "TEMNOME"]
self.assertListEqual(test, expected)
def test_ibge_parser(self):
df = ibge_json_to_df(get_sample_json("ibge_json.json"))
# assert it is a DataFrame
self.assertIsInstance(df, pd.DataFrame)
# index is datetime
self.assertTrue(pd.api.types.is_datetime64_dtype(df.index))
def test_ipea_metadata_to_df(self, mocked_get_json):
mocked_get_json.return_value = get_sample_json("ipea_metadata.json")
df = ipea.ipea_metadata_to_df("url")
self.assertFalse(df.empty)
def generate_jsons():
"""Simulate making a IPEA API response, then a BCB API response"""
srcs = ["ipea", "bcb"]
for src in srcs:
yield get_sample_json(src + "_json.json")
def setUp(self):
mock_json(
path="seriesbr.helpers.searching.get_json",
json=get_sample_json("ipea_search_results.json"),
).start()
