| from django.test import TestCase |
| from core.models import MutualFund, Stock |
| from core.mfrating.score_calculator import MutualFundScorer, MFRating |
| from core.tests.data import test_data |
|
|
|
|
| class MutualFundScorerTestCase(TestCase): |
| """ |
| Test case for the MutualFundScorer class to test scores. |
| """ |
|
|
| def setUp(self): |
| self.stock_data = [ |
| {"isin_number": "INE040A01034", "rank": 10}, |
| {"isin_number": "INE090A01021", "rank": 21}, |
| {"isin_number": "INE002A01018", "rank": 131}, |
| {"isin_number": "INE154A01025", "rank": 99}, |
| {"isin_number": "INE018A01030", "rank": 31}, |
| {"isin_number": "INE280A01028", "rank": 2}, |
| ] |
|
|
| self.mutual_fund_data = [ |
| { |
| "isin_number": "ISIN1", |
| "fund_name": "Testing Fund 1", |
| "rank": 1, |
| "aum": 837.3, |
| "crisil_rank": 4, |
| "security_id": "SEC1", |
| "data": test_data[1], |
| }, |
| { |
| "isin_number": "ISIN2", |
| "fund_name": "Testing Fund 2", |
| "rank": 2, |
| "aum": 210.3, |
| "crisil_rank": 1, |
| "security_id": "SEC2", |
| "data": test_data[2], |
| }, |
| { |
| "isin_number": "ISIN3", |
| "fund_name": "Testing Fund 3", |
| "rank": 3, |
| "aum": 639.3, |
| "crisil_rank": 3, |
| "security_id": "SEC3", |
| "data": test_data[3], |
| }, |
| { |
| "isin_number": "ISIN4", |
| "fund_name": "Testing Fund 4", |
| "rank": 4, |
| "aum": 410.3, |
| "crisil_rank": 2, |
| "security_id": "SEC4", |
| "data": test_data[4], |
| }, |
| { |
| "isin_number": "ISIN5", |
| "fund_name": "Testing Fund 5", |
| "rank": 5, |
| "aum": 1881.3, |
| "crisil_rank": 5, |
| "security_id": "SEC5", |
| "data": test_data[5], |
| }, |
| ] |
|
|
| self.create_stock_objects() |
| self.create_mutual_fund_objects() |
| self.mf_scorer = MutualFundScorer() |
|
|
| def create_stock_objects(self): |
| """ |
| Create stock objects using the predefined stock data. |
| """ |
| self.stock_objects = [Stock.objects.create(**data) for data in self.stock_data] |
|
|
| def create_mutual_fund_objects(self): |
| """ |
| Create mutual fund objects using the predefined mutual fund data. |
| """ |
| self.mutual_fund_objects = [ |
| MutualFund.objects.create(**data) for data in self.mutual_fund_data |
| ] |
|
|
| def test_get_scores_returns_sorted_list(self): |
| """ |
| Test whether the get_scores method returns a sorted list of scores. |
| """ |
| scores = self.mf_scorer.get_scores() |
| self.assertEqual(len(scores), 5) |
| self.assertEqual( |
| scores, sorted(scores, key=lambda x: x["overall_score"], reverse=True) |
| ) |
| expected_scores = [0.4263, 0.3348, 0.2962, 0.2447, 0.2101] |
| for i, expected_score in enumerate(expected_scores): |
| self.assertAlmostEqual( |
| scores[i]["overall_score"], expected_score, delta=1e-4 |
| ) |
|
|
