feat: update ETF portfolio deposit file structure and tests for new columns

Author: seohyunjun

pykrx/website/krx/etx/wrap.py

@@ -210,18 +210,18 @@ def get_etf_portfolio_deposit_file(date: str, ticker: str) -> DataFrame:
 
     Returns:
         DataFrame:
-                     계약수       금액       비중
-            티커
-            005930   8175.0  694875000  16.531250
-            000660    972.0  126360000   2.949219
-            051910     80.0   77120000   1.849609
-            035420    219.0   65809500   1.570312
+                구성종목명	계약수	금액	시가총액	비중
+            티커					
+            0724F1	ADOBE SYSTEMS INC	12.22	0	0	0.0
+            2079K3	ALPHABET INC-CL A	61.55	0	0	0.0
+            2079K1	ALPHABET INC-CL C	57.21	0	0	0.0
+            255371	AMERICAN ELECTRIC POWER	15.61	0	0	0.0
     """
 
     isin = get_etx_isin(ticker)
     df = PDF().fetch(date, isin)
-    df = df[["COMPST_ISU_CD", "COMPST_ISU_CU1_SHRS", "VALU_AMT", "COMPST_RTO"]]
-    df.columns = ["티커", "계약수", "금액", "비중"]
+    df = df[["COMPST_ISU_CD", "COMPST_ISU_NM","COMPST_ISU_CU1_SHRS", "VALU_AMT", "COMPST_AMT","COMPST_RTO"]]
+    df.columns = ["티커", "구성종목명", "계약수", "금액", "시가총액", "비중"]
 
     # NOTE: 웹 서버가 COMPST_ISU_CD에 ISIN과 축향형을 혼합해서 반환한다. Why?
     df["티커"] = df["티커"].apply(lambda x: x[3:9] if len(x) > 6 else x)
@@ -231,7 +231,7 @@ def get_etf_portfolio_deposit_file(date: str, ticker: str) -> DataFrame:
     # - empty string은 int, float로 형변환 불가
     #  -> 이 문제를 해결하기 위해 '-' 문자는 0으로 치환
     df = df.replace(r"\-$", "0", regex=True)
-    df = df.astype({"계약수": np.float64, "금액": np.int64, "비중": np.float32})
+    df = df.astype({"계약수": np.float64, "금액": np.int64, "시가총액": np.int64, "비중": np.float32})
     df = df[(df.T != 0).any()]
     return df
 

tests/integration/test_etf_api.py

@@ -148,13 +148,13 @@ class TestEtfPdf:
     @pytest.mark.vcr
     def test_with_business_day(self):
         df = stock.get_etf_portfolio_deposit_file("152100", "20210402")
-        temp = df.iloc[0:5, 0] == np.array([8140.0, 968.0, 218.0, 79.0, 89.0])
+        temp = df.iloc[0:5, 1] == np.array([8140.0, 968.0, 218.0, 79.0, 89.0])
         assert temp.sum() == 5
 
     @pytest.mark.vcr
     def test_with_negative_value(self):
         df = stock.get_etf_portfolio_deposit_file("114800", "20210402")
-        assert df.iloc[1, 0] == pytest.approx(-3.58)
+        assert df.iloc[1, 1] == pytest.approx(-3.58)
 
 
 class TestEtfTradingvolumeValue: