From 01b82a03bc596c626da40a663d56c0f3f05aab4e Mon Sep 17 00:00:00 2001
From: "google-labs-jules[bot]"
 <161369871+google-labs-jules[bot]@users.noreply.github.com>
Date: Wed, 11 Mar 2026 18:37:47 +0000
Subject: [PATCH] Add test file for data/data_loader.py

This commit introduces a new test file, `tests/test_data_loader.py`, to verify the functionality of `get_data_mt5`. It uses `unittest.mock.MagicMock` to mock the `MetaTrader5` module, allowing tests to run on Linux where the module cannot be imported.

The tests cover:
- Live trading data retrieval (`start_pos=None`).
- Backtesting data retrieval with a specific `start_pos`.
- Error handling when `copy_rates_from_pos` returns `None`.
- The proper conversion of rates into a pandas DataFrame with a parsed datetime index.

Co-authored-by: maghdam <63883156+maghdam@users.noreply.github.com>
---
 data/__pycache__/__init__.cpython-311.pyc    | Bin 164 -> 0 bytes
 data/__pycache__/__init__.cpython-312.pyc    | Bin 159 -> 0 bytes
 data/__pycache__/data_loader.cpython-311.pyc | Bin 1660 -> 0 bytes
 data/__pycache__/data_loader.cpython-312.pyc | Bin 1566 -> 0 bytes
 tests/test_data_loader.py                    |  74 +++++++++++++++++++
 5 files changed, 74 insertions(+)
 delete mode 100644 data/__pycache__/__init__.cpython-311.pyc
 delete mode 100644 data/__pycache__/__init__.cpython-312.pyc
 delete mode 100644 data/__pycache__/data_loader.cpython-311.pyc
 delete mode 100644 data/__pycache__/data_loader.cpython-312.pyc
 create mode 100644 tests/test_data_loader.py

diff --git a/data/__pycache__/__init__.cpython-311.pyc b/data/__pycache__/__init__.cpython-311.pyc
deleted file mode 100644
index 8f779c6403d55137c5917b285049c6ff6b4b5451..0000000000000000000000000000000000000000
GIT binary patch
literal 0
HcmV?d00001

literal 164
zcmZ3^%ge<81R;kvri19mAOZ#$p^VRLK*n^26oz01O-8?!3`I;p{%4TnFY{!pn9$<X
zqT-m`{EWoh82`Lfm!izF)R^3y_@w-j_>!W;l+3*Jn3Tkl#F+T_%)HE!_;|g7%3mBd
ix%nxjIjMF<tUz->HWl*&i4V+-jEo-`U_=o!Pz(V6&?nab

diff --git a/data/__pycache__/__init__.cpython-312.pyc b/data/__pycache__/__init__.cpython-312.pyc
deleted file mode 100644
index e31893c5dbc0970e71b4e5bf359d7d46dbc233e9..0000000000000000000000000000000000000000
GIT binary patch
literal 0
HcmV?d00001

literal 159
zcmX@j%ge<81R;kvri19mAOanHW&w&!XQ*V*Wb|9fP{ah}eFmxdWtVIf6Iz^FR2-9=
zpOKgw<DZx6Qj}Si8snH#kdf$?lV7e7Qk0mInU}8MlwT5)l30=$6Ca<Mmst`YuUAlc
gi^C>2KczG$)vkyYXdELD7lRldnHd=wi<p5d0Bl+(nE(I)

diff --git a/data/__pycache__/data_loader.cpython-311.pyc b/data/__pycache__/data_loader.cpython-311.pyc
deleted file mode 100644
index 447eec9936b80bccacc0e07b2df5034d9b71cfbf..0000000000000000000000000000000000000000
GIT binary patch
literal 0
HcmV?d00001

literal 1660
zcmZ`(O=ufO6rR=oNV05KvPvx%w@gEDEl349sL&AF%CSrxEC(8=2V;=enptVx{SmXX
zPF-YBL!mzOkVB!RrQkzxTarVLhF%K2^w5I@I*0{BpyZI7!MB|HW+kmFjEC7bZ)V>6
z-tX+MQ&STN#z)uxs{f%N^sm^AMtsPe-v#D-L=cfIv?+yaY*P+bWmAPUW@)yzsY!^$
ziA>`}p$Vc=c{e$NGBl!nt!_>b=|0LQUW()Oyc8se&M7x-8ff0Wz2jPe=2Uld<^>v~
z+-FV=e2Rt!oDARjbxHWslG`~B+-U4Zd<(T=B-WCSLHkLBaG*G|SIf8L%knLyJ;T)z
zK0-upp`Xytv173GB(nBM;-gG~=x$`MrHsoRsAF;wZc8P~Tx5i(PqpJ1;KMf<;{G<U
zE%iub;L;!pi9bz@>qsCCnNI32!OaN_V}<fs12;^MyUeWV7KYU8*kG=WE0pUSOed7#
zVj&a`$$NNvP?YUkSX`S<9h;8FnQv2v<3(Di7jEEdOB;6{++V(yA83Nopn{;Z?%O+n
z=^9v+6mu6tQJG2iXe29YZ!q1aAYR!h;%GmDw|TL!v05pwd|IlMtDoGwFynEZ@oLlc
zK=qzsc$5#K;SGvy*W+NUCj123c#%*;_braM*IkEhW5Z?GGWRIvgSdvMVc4q%#2sN{
z(;F(?z=lX6g+Rc91&s?fM#17m-=l_aU4?uZv<~K-3(N>8tO#%D{QtFrC{1p<&X}1U
zz4p0q0_rmKf1Qb@`Eve_>stgnuE@b4;lpY}*ntwhH_3zNr}&P^H-L$oPSetBG#?LS
z$as*gxy}756ERkWLn6(=#AmwY(=uZ&3t~+YB$uIsRz&fF6nCr9Q%*uDs-{Ee7kQ0|
z*T6)p2g#_mK{Cp3poT_*7%@E2+@e!4epRa7+Irwo=55(-L$|l?Idqvp!nR6Qv!Sn8
z?!%}fmt4LjS~(<D%N1>0XzmA-b?`!5R3RXi1MCER-j6`q=q!g4nNB=+n$Gsp@BXGe
zOP<_3F}mq;FI_%V57ieJ=O6qD=apW%a;QF!Pqq(Er*oaTk54opFuT*G-gK#xD7{Fg
zzMX$eyUF=pa=yKEn#q3m_EGHU{a+@YWjmQ-H&g6oitVNL-dA_~iuTRgqqWCx9KGIN
z>naO9Wuc=i^ixyq)pPI;<)!s}Do9<t$|9SD<3MZb4$(dKn%Jn0Z$p$Zr^&=GF_7V1
z2D0gJmWFK@@9&6--wK-r@+xuLc8PD%57|4g75^^pA&@g!lB6>YNpE)0++g-mwlkW2
mG}9T)K6<M&KF^d{Y5EKS`5_xFN40SId*+|{khzkCj{OHy3cwHm

diff --git a/data/__pycache__/data_loader.cpython-312.pyc b/data/__pycache__/data_loader.cpython-312.pyc
deleted file mode 100644
index 840903be11af372ba60d2278c261dcff064542cb..0000000000000000000000000000000000000000
GIT binary patch
literal 0
HcmV?d00001

literal 1566
zcmah}PiP!f7=N?#Xa6N}la+R(Yk8JpvrrduSjAG*O_N|Gn?k$wuq<pQ``+wMoIlH(
zH?ffoY(>zVtfvYh#G8^HatwG8yeJ-wq+ppwDA-GHk$Cdt_hxr?6GVJ4@B7|+-}n3e
z&(GuIDFo}SOFy)}QxN(?9HJE;Fef*GxsM1Ul7(tgNMkiQq)JT%8nZN8t7#G<aU#<=
zQD}mwR9>Hophncti1x8sOKnPp#4~ZdQjmfK(K+R&O#{u_wAWlK(45Aa&b&Zll>5wS
zf^Wm60Vl(Ea!wL{b>-eP2d=-?V-Yu^UDS;csVnUcP_`5&_x~DQ`Lsq?=}mAo!X&XS
z`VxH=+a1{bNa9hp(0U@08quy2Np7j9B*WZ~qy*Phi98LIC=b-#K`i3#TKKj;E@aLw
zA`|UFJd%|F;T9?+^k?8&3d35YyxGQW)8j5Po4SP|**Z3uYvT&#dX?#fGF&Q#!r}HR
zUhS7*^(q$6rqjZv<8kKO)ZuuB7F)$D_~Kmk`t4iu7Ylt&Q0mtVlvaFu4KQ5;iy~s~
zVkjds=>{FjifS87w<(BMswF%;NANZ;6{|~?g~glY%0lCv*^wEK>x?%#t_P|&48x<m
z9}RC)Y`Y!@V@=^F*v2!28oF<Byt?8#bQK#e!<M;0G4IDUKn=rQG9d0DHa5M1(iLoo
z6jBHTELhO0U?U0^&-flSeCxlE{|(v#^Uer!2ozR?w{?EHRuHAhP1iYNW=(ItFPwn7
z4E$%JQ$EaJcYTXs#}z5)=X+3Wh&oWh0h2tKH^p~MUIiv@Ivq=I(n36tA>l!`>2@|7
zOoZ4F{)j9GsrPitrwfd^EQobTker7GS`@Vl(%fwfXEzCzXqXP6cM2L4qsGL%1j%7(
zgXA#1ff^bKV#M&oJc~xf`0sM_YW=oHnOC>nwr<yNICP#tzUpPG)7BR)_k&?W&bhoU
zIyu}LmMglr*x3xmTHu9vXh1+L2iRgBy=~Zf=s1TGlLztKQ6~Fa=JKzZ%im~w$^F@V
z<1n+Ztv)?-`^NWhTlrLd5+Cbr9Zlp8UVMFD+q-ix{r2HR`5;k#noNI|-_GBohsk_z
z?r1W5|H9`Nc49lPeVy9NKAbG|=6V}<-+iQLpDf>9zISft`QGwF<&{V2@!rx2ID-n(
zN+BJjM>AMrV{jd49o-?i$DR`h)$wfzD&};U_~ivMOkp6K4rduShT(mSnD}9^^Wr^m
z&31`z(Kp%4a1?(hZx*&=S(2m^4M{INM%l+`>UZ?gpUQb@;*0Dt0<zN#$&XXN<bTTl
JC2m5<e*lF`qagqQ

diff --git a/tests/test_data_loader.py b/tests/test_data_loader.py
new file mode 100644
index 0000000..8f34001
--- /dev/null
+++ b/tests/test_data_loader.py
@@ -0,0 +1,74 @@
+import sys
+from unittest.mock import MagicMock
+
+# Mock MetaTrader5 before importing data_loader
+mt5_mock = MagicMock()
+sys.modules['MetaTrader5'] = mt5_mock
+
+import pandas as pd
+import pytest
+from data.data_loader import get_data_mt5
+
+def test_get_data_mt5_live_trading():
+    """Test get_data_mt5 when start_pos is None (live trading)."""
+    # Arrange
+    symbol = "BTCUSD"
+    n_bars = 100
+    timeframe = mt5_mock.TIMEFRAME_H1
+
+    # Mock return value of copy_rates_from_pos
+    mock_rates = [
+        {"time": 1600000000, "open": 1.0, "high": 2.0, "low": 0.5, "close": 1.5},
+        {"time": 1600003600, "open": 1.5, "high": 2.5, "low": 1.0, "close": 2.0},
+    ]
+    mt5_mock.copy_rates_from_pos.return_value = mock_rates
+
+    # Act
+    df = get_data_mt5(symbol, n_bars, timeframe)
+
+    # Assert
+    mt5_mock.copy_rates_from_pos.assert_called_once_with(symbol, timeframe, 0, n_bars)
+    assert isinstance(df, pd.DataFrame)
+    assert df.index.name == 'time'
+    assert len(df) == 2
+    assert "open" in df.columns
+    assert df.index[0] == pd.to_datetime(1600000000, unit='s')
+
+def test_get_data_mt5_backtesting():
+    """Test get_data_mt5 when start_pos is provided (backtesting)."""
+    # Arrange
+    mt5_mock.copy_rates_from_pos.reset_mock()
+    symbol = "EURUSD"
+    n_bars = 50
+    timeframe = mt5_mock.TIMEFRAME_M15
+    start_pos = 10
+
+    mock_rates = [
+        {"time": 1600000000, "open": 1.1, "high": 1.2, "low": 1.0, "close": 1.15},
+    ]
+    mt5_mock.copy_rates_from_pos.return_value = mock_rates
+
+    # Act
+    df = get_data_mt5(symbol, n_bars, timeframe, start_pos=start_pos)
+
+    # Assert
+    mt5_mock.copy_rates_from_pos.assert_called_once_with(symbol, timeframe, start_pos, n_bars)
+    assert isinstance(df, pd.DataFrame)
+    assert df.index.name == 'time'
+    assert len(df) == 1
+
+def test_get_data_mt5_no_data():
+    """Test get_data_mt5 when copy_rates_from_pos returns None."""
+    # Arrange
+    mt5_mock.copy_rates_from_pos.reset_mock()
+    symbol = "INVALID"
+    n_bars = 10
+    timeframe = mt5_mock.TIMEFRAME_H1
+
+    mt5_mock.copy_rates_from_pos.return_value = None
+
+    # Act & Assert
+    with pytest.raises(ValueError, match=f"Could not retrieve data for {symbol}"):
+        get_data_mt5(symbol, n_bars, timeframe)
+
+    mt5_mock.copy_rates_from_pos.assert_called_once_with(symbol, timeframe, 0, n_bars)