def test_entity_id(self, requests_mock):
requests_mock.get(
"https://petermaffay.addepar.com/api/v1/entities/1",
json={"data": {
"id": "1",
"attributes": {
"a": 2.0,
"b": 3.0
}
}})
x = Request().entity(id=1)
assert x == {"a": 2.0, "b": 3.0}
def test_view_csv(self, requests_mock):
xxx = pd.DataFrame(index=["x", "y"], columns=["a"], data=[[4], [5]])
xxx.index.name = "Peter"
requests_mock.get(
"https://petermaffay.addepar.com/api/v1/portfolio/views/10/results?portfolio_id=20&portfolio_type=entity&output_type=csv&start_date=1978-11-12&end_date=1978-11-12",
text=xxx.to_csv())
x = Request().view_csv(view_id=10,
portfolio_id=20,
portfolio_type=PortfolioType.ENTITY,
start_date=pd.Timestamp("12-Nov-1978"),
end_date=pd.Timestamp("12-Nov-1978"))
pdt.assert_frame_equal(x, xxx.reset_index())
def test_entities(self, requests_mock):
requests_mock.get("https://petermaffay.addepar.com/api/v1/entities",
json={
"data": [{
"id": "1",
"attributes": {
"a": 2.0,
"b": 3.0
}
}],
"links": {
"next": None
}
})
x = {key: attr for key, attr in Request().entities}
assert x == {"1": {"a": 2.0, "b": 3.0}}
def test_version(self, requests_mock):
r = Request(key="1", secret="1", id="maffay", company="maffay")
requests_mock.get("https://maffay.addepar.com/api/v1/api_version",
json={'a': 'b'})
y = r.version
assert y == {'a': 'b'}
def test_dicturl(self):
x = Request().dicturl({"a": 200, "b": "cc"})
assert x == "a=200&b=cc"
def test_auth(self):
r = Request(key="a", secret="bb")
assert r.auth == ("a", "bb")