class TestMLOrderRecommender(unittest.TestCase):
def setUp(self):
self.order_recommender = MLOrderRecommender(
note_model=DummyModel(), authorization="topsecret", investor_id="123"
)
@httpretty.activate
def test_get_model_rec(self):
httpretty.register_uri(
method=httpretty.GET,
uri="/".join([API_CONFIG["base_url"], API_CONFIG["version"], "loans/listing"]),
body=json.dumps({"loans": [{"id": "l1"}, {"id": "l2"}, {"id": "l3"}]}),
)
rec = self.order_recommender.get_model_rec()
self.assertEqual(rec, {"l1": {"amount": 25}, "l2": {"amount": 25}})
@httpretty.activate
def test_execute_recommended_order(self):
expected_body = [
{
"orderInstructId": 10,
"loanId": "l1",
"requestedAmount": 25,
"investedAmount": 25,
"execution_status": "ORDER_FULFILLED",
},
{
"orderInstructId": 10,
"loanId": "l2",
"requestedAmount": 25,
"investedAmount": 25,
"execution_status": "ORDER_FULFILLED",
},
]
httpretty.register_uri(
method=httpretty.GET,
uri="/".join([API_CONFIG["base_url"], API_CONFIG["version"], "loans/listing"]),
body=json.dumps({"loans": [{"id": "l1"}, {"id": "l2"}, {"id": "l3"}]}),
)
httpretty.register_uri(
method=httpretty.POST,
uri="/".join([API_CONFIG["base_url"], API_CONFIG["version"], "accounts/123/orders"]),
body=json.dumps(expected_body),
)
response = self.order_recommender.execute_recommended_order(portfolio=0, safe_mode=False)
self.assertEqual(response.json(), expected_body)
self.assertEqual(response.status_code, 200)
def setUp(self):
self.order_recommender = MLOrderRecommender(note_model=DummyModel(),
authorization='topsecret',
investor_id='123')
class TestMLOrderRecommender(unittest.TestCase):
def setUp(self):
self.order_recommender = MLOrderRecommender(note_model=DummyModel(),
authorization='topsecret',
investor_id='123')
@httpretty.activate
def test_get_model_rec(self):
httpretty.register_uri(method=httpretty.GET,
uri='/'.join([
API_CONFIG['base_url'],
API_CONFIG['version'], 'loans/listing'
]),
body=json.dumps({
"loans": [{
'id': 'l1'
}, {
'id': 'l2'
}, {
'id': 'l3'
}]
}))
rec = self.order_recommender.get_model_rec()
self.assertEqual(rec, {'l1': {'amount': 25}, 'l2': {'amount': 25}})
@httpretty.activate
def test_execute_recommended_order(self):
expected_body = [{
"orderInstructId": 10,
"loanId": "l1",
"requestedAmount": 25,
"investedAmount": 25,
"execution_status": "ORDER_FULFILLED"
}, {
"orderInstructId": 10,
"loanId": "l2",
"requestedAmount": 25,
"investedAmount": 25,
"execution_status": "ORDER_FULFILLED"
}]
httpretty.register_uri(method=httpretty.GET,
uri='/'.join([
API_CONFIG['base_url'],
API_CONFIG['version'], 'loans/listing'
]),
body=json.dumps({
"loans": [{
'id': 'l1'
}, {
'id': 'l2'
}, {
'id': 'l3'
}]
}))
httpretty.register_uri(method=httpretty.POST,
uri='/'.join([
API_CONFIG['base_url'],
API_CONFIG['version'], 'accounts/123/orders'
]),
body=json.dumps(expected_body))
response = self.order_recommender.execute_recommended_order(
portfolio=0, safe_mode=False)
self.assertEqual(response.json(), expected_body)
self.assertEqual(response.status_code, 200)
def setUp(self):
self.order_recommender = MLOrderRecommender(
note_model=DummyModel(), authorization="topsecret", investor_id="123"
)
