From 192be8c6f9d89e667e3148476d3eb2256aefc672 Mon Sep 17 00:00:00 2001
From: mmasoud1 <mmasoud2@outlook.com>
Date: Mon, 10 Feb 2025 03:38:28 -0500
Subject: [PATCH] Add testing file

---
 tests/__pycache__/__init__.cpython-38.pyc     | Bin 0 -> 189 bytes
 .../test_dice.cpython-38-pytest-6.2.4.pyc     | Bin 4270 -> 4270 bytes
 .../test_meshnet.cpython-38-pytest-6.2.4.pyc  | Bin 0 -> 1797 bytes
 tests/test_meshnet.py                         |  34 ++++++++++++++++++
 4 files changed, 34 insertions(+)
 create mode 100644 tests/__pycache__/__init__.cpython-38.pyc
 create mode 100644 tests/__pycache__/test_meshnet.cpython-38-pytest-6.2.4.pyc
 create mode 100644 tests/test_meshnet.py

diff --git a/tests/__pycache__/__init__.cpython-38.pyc b/tests/__pycache__/__init__.cpython-38.pyc
new file mode 100644
index 0000000000000000000000000000000000000000..6be8d5c43642d285600882284886fb4e9c5e7a4c
GIT binary patch
literal 189
zcmWIL<>g`kg3hHY(?RrO5P=LBfgA@QE@lA|DGb33nv8xc8Hzx{2;x_ger{??W}<#>
zenujYrEhF%W}K8{YNl&pVVS6Fl9ZUDn`mNTrkj#zo@SJ0kYtu<V65+3>7H4VQJSRh
zoS&IjT#}d^pO}-Q@0(hj;pU@Xl3H9+tREkrnU`4-AFo$Xd5gm)H$SB`C)EyQ_h%qx
F008B0E;#@I

literal 0
HcmV?d00001

diff --git a/tests/__pycache__/test_dice.cpython-38-pytest-6.2.4.pyc b/tests/__pycache__/test_dice.cpython-38-pytest-6.2.4.pyc
index 853fdf4c1283f11cd4697d72d1943e14f08babf4..c7ad4670c0fc4c82e233cead694c6d988d2dd49f 100644
GIT binary patch
delta 19
ZcmZ3dxK5EPl$V!_0SJ<pZsb}b001qa1epK;

delta 19
ZcmZ3dxK5EPl$V!_0SGF4H*zfz001qm1eX8+

diff --git a/tests/__pycache__/test_meshnet.cpython-38-pytest-6.2.4.pyc b/tests/__pycache__/test_meshnet.cpython-38-pytest-6.2.4.pyc
new file mode 100644
index 0000000000000000000000000000000000000000..b97588e99546ee7d7a15325993dbfa5aa4613d38
GIT binary patch
literal 1797
zcma)6&2Jk;6rb5Iuh)(nJEd&|2upy(0<lBB8Y)t?ssa*fEAb_eFRSs+I$meLG_x*A
zSWAd<Mf8YNkZi=I7tWkH_b-?$r|<`Gs5tOu)(%18&|U4@H}C!C&HMdkc0XHMau8fk
ze|vK9qKVKSa&Z3AU~mf_{vjBOD2`AMtG5>G8p0?xdL}M2R?kwIO0S~ccF%^l7FA=X
z=Kw~h#<!?fqxci#nx{|?-FLBQ2FbmE4?TYv_=i~<Cb=(Di>n4s6`nir@V_uL0n0~d
zf^t04rn<65fkx)Y8dYYPYBOzub6YM(%>q;FJ7c2d)u{s!omS*MGbehXO=~k~zt9T(
zA)Xl2o|uI(s!y!KD$GM|QYk8rFt_uiX@gda$^cVGW(L~Kptbzcba{pjE&c=Kn}spM
z>}T*=(n~bROW%TS>SxH`719LcQ<Vh|kafEBqkaWV<T@*~QJ9Yry(Ig1jDCVzbonZx
z&B6{bV5<3LK)p~@XZRuh22?Q6#DYGx=@rQYI9{Eh?;B^FoX^oV`UriDzCeX@f!Uu>
z>msw(qF><t01(Z>nc+v6eGPs~`p|Pr`Tt=D=wB2XSONN1h4?>)D|3d>&+{{+2oB8U
z`wTW8oz&x$2GQ;P&WNYUsSKU!(!k+Gt|4k+0w;+iNf2>S1wSI32VB^TV6Wz<$sj!N
z24NJOF3VXalNFdDjXY}q8W%s)_V!$SviABL+3|+Uov-Fjd#^3Mb(eSekYs@~Y44Sx
zEtiXWI?l6k?&XYxiD(2zS>WdZ^%T_Ag$^^N*^p-I<gHI7t}U6h%b9J3Y{wywN$wAC
zv==4XC)V5TgEViSEIT(PHI}PJc3@SgwIgctrGp0~I(c&+?DN%iul1TF%Q`+Z(Fu+M
zf1IaGXek#)M)ILBM(}Bh3JqC8;y@S@C+stz?I}S$kI6rVFk!Mk6ec4HO<<&SFcm4`
z+{%vgfae}zqUrf0iaZu%%unNtut3xy?e)iDl!plyR+IvQi))}=I*zD<dIwS3Cy}Q*
zsd}6Y0$Bm74N?Yzd8IHhKM;-EN~17M_89|0D+>q+8j*zLkP(IyOH`87J79#mHKBtN
zz2$QU#ZwxX>g)-8#N=Oz)g{xKY`0wRo!+^@W;rMu+Ad+`;5uu<#3zWY!Ry(e8wWHb
z-8dbBuC%+ky|dZxZ||(_?!HUbw)%vwk*(dGHA=1zHU{hc9kRaJy?6XUm=DMO?j6_^
zo)g~#6}$2``u^Rn)Rrsj#Slq?yptV^SC!=RkH;IZH1;x<(y=cU@Jgx{G{kR!LAs4!
z(ltE4|F-NFL~Lwe3)#p*L-ZPIARA&1Zp~wvim@)tcI(1k>>zsu0%FxG_q<ao1xt2V
z@FffLKvd5{qB4)MmthWTNwX?vTjk{5>V*T^nKM+UPOYs}J!d+yfc+j2A4kDW`M=<D
N-)z)El97$;+CMew3akJC

literal 0
HcmV?d00001

diff --git a/tests/test_meshnet.py b/tests/test_meshnet.py
new file mode 100644
index 0000000..af24076
--- /dev/null
+++ b/tests/test_meshnet.py
@@ -0,0 +1,34 @@
+import os
+import torch
+import pytest
+from app.code.executor.meshnet import enMesh_checkpoint
+
+def test_modelAE_from_production_config():
+    # Emulate production: get the modelAE.json from app/code/executor/
+    # Import the package to use its __file__ attribute.
+    import app.code.executor  
+    config_file_path = os.path.join(os.path.dirname(app.code.executor.__file__), "modelAE.json")
+    
+    # Instantiate the model using production parameters.
+    model = enMesh_checkpoint(
+        in_channels=1,
+        n_classes=3,
+        channels=5,
+        config_file=config_file_path
+    )
+    
+    # Create a small random input tensor.
+    x = torch.randn(1, 1, 16, 16, 16)
+    
+    # Test in train mode (which uses checkpoint_sequential)
+    model.train()
+    output_train = model(x)
+    expected_shape = (1, 3, 16, 16, 16)  # Expecting n_classes=3 as output channels.
+    assert output_train.shape == expected_shape, f"Train mode output shape mismatch: expected {expected_shape}, got {output_train.shape}"
+    
+    # Test in eval mode (using inference mode)
+    model.eval()
+    with torch.no_grad():
+        output_eval = model(x)
+    assert output_eval.shape == expected_shape, f"Eval mode output shape mismatch: expected {expected_shape}, got {output_eval.shape}"
+