From fdc596d1d5a7c36c440f32e45618b66fe80cdb41 Mon Sep 17 00:00:00 2001
From: kasinadhsarma <kasinadhsarma@gmail.com>
Date: Thu, 26 Dec 2024 13:30:37 +0530
Subject: [PATCH] Add environment variable checks to test suite and include
 numpy as a required package

---
 ...t_environment.cpython-310-pytest-8.3.4.pyc | Bin 9452 -> 9898 bytes
 tests/test_environment.py                     |  14 ++++++++++++--
 2 files changed, 12 insertions(+), 2 deletions(-)

diff --git a/tests/__pycache__/test_environment.cpython-310-pytest-8.3.4.pyc b/tests/__pycache__/test_environment.cpython-310-pytest-8.3.4.pyc
index 109ddada61d6b735089008e1d8b2412bd405f02f..50c3dc702185fd94e23256316853ee4a41190d81 100644
GIT binary patch
delta 2243
zcmaJ?U2Gdg5WYR%*%#Mw6Wd9g#7UgLCRR#XCxt>A6{<G0d1y+1(n8eYTzp&S9RFqS
zY#KLBphZH07SXPdN=qSmLhx4k6%Z0HeL|IhmlN$H0#PB5;05tFvrdyXRQPnC=VoVj
zznR&cxj**FA%DKM)+@p9Cto&ovG0mMLhfA|D4sGTgJc4e0a}v|!W%Nw%Y;a@X1Rej
zu8_&51Crr3Jg-ZJhkDZCi>}FL+GuFAGRxAY<rbr61+7u?Qf*mVk#0ygWufz=Td(|<
zAQ$*?a?WRjUza9>M)NT#*22Fg`jaX!*$1z+1;8MN46n;j-d7f+gk-3OJ0*ie7Nli)
zMIx{}%+&a4x!+rM#|MYvgWF>Ye?yM$cGO%kTbOs0gNF}|I?4-sj~^V5xtw}N=ZYqy
zMxvl6XY~}d9#YX<DN)iH|5T2V0RLXrn*vD14<IWt!AJYk>-Le#r0Y5%e*U{MOX|6<
zwujJ=2?(N82hd*O>KGv%{H}Wo*}@f1Z_lGvPSB>7ohZ!PGx=PiM44sgbMeX{&o2sT
zsqFE^adM%yc?ZUV^#M4(qs5$UX6YzndB%ECrRkPMnLU<cLuiI%W1HD9|E%`%jsev4
zqm?_GH;NfrrMSk^sTT~Ac}5dvwvcDG^-N{aKO}?ux9XngS~v8LWK7FW=Cg%8DOk^v
z0m4#wgzKJ0-X9F`{`&986juX%Zozhl4+XkYM{#SF3bKbm7ETD-Y#csg9yW{?{$$b6
z6M9KEGx{`;%%IX|(2|*?iJYFL7Bf&q&NAVVphZE$FxXBLudi<+FY|8#uacknL~xfX
z8)}O1tHHeu`xi)>;Q1rcNh#N*NOsM9ivJLtaP`i0^5;U&2hbgc5R-qr;{o{<el7HD
z_YA7VM$>E3S~%Dl&|TnvgbuctAe95hX7mD$W=$)r+sT<|J{M(E{8Ym>(!wt_^o&p8
z60-<d1oV&P0h~?M<c-c1^^BuhxC$%aD5h4#h>5sWV+y*M2C4nKp*@<$>565RA@^vz
z9+)aiKGN9gKMy0~CTX%J4_0X7T>^~uh2ynbaV?qQAKS`b3-5JLp*YKLhuhs%lKg(S
zm9A}EJ%1|Gb4CtMlWu2=88(k|$_S70T}>?0a7d)imDCG@s6*!e%Qe&7N#3cv-TWod
z&cc+BWtQeIwT=ynB)kQh7;!vjGI`w|9%3RNBKwc#;}*ZuI<cvQTVin7CB8iphBG-9
ziG$PUA_J3`LA|=pQ#;rc8evi#H$7|9oUtJ}RqrmL)&HkNWy4q7A`KhTAQ%RDysd|P
zR5{T$OKS0II;v^Yti>YytM-<rI4-Ut><5S`;ySQ?{!e?XPB^#bp%dYY9TD<HWuoI(
zQgsp|#Y}#r^WuwmmNq6sWNr=z#0kRtw(c@J3ZD=56It58zPd5r+SNmD^CMm9r6hqF
z4}LBX;Hp$BHBw)Y($aqEbmTm-$+EmaR>*m%)61>}sNW0ZhGfV$T`Mw?Y-L$p)xa3^
z1O<D-98`Fy<7#>E1ZC!#`6z{w26ax)M4=>`ET79R!FLpDvxx=5it3C;Eozq=M_0!m
z9**{veSJ~WistgRuyXb7ijK$JY%?%*<h<p$Qxp<jV()^2tsU{0t6Gj+u#cmb^h|NR
zUhZ72H7%lGK41mAV#BDCx(AC{*h;;!mxM@|D1Vg)us((#Yd#(e@z1+<FNKf;_6xQX
zA%UPEh_p<hBq~P{rA34{5w0R!LwFD20|b$@br78h(}aFoQ2<R+hz#uHO<tk}G_9`J
zTT8TdEvPkV&6?(G<cr-;cZk|RoEqpz(|OCZ=VK8^n=YCe+ss+)0ROgo*gXoXu@3I<
z=_GA@d(SqRUFWBIx(Xkn9wX{#$Dz6H)1fyZ1<{-wC99jc$I*1Ok!pk5gU3vitS(6-
Y)u)d5;6d3B+yk<-q@BbF*Fl8;1zbu4vH$=8

delta 1781
zcmaJ>T}&KR6rMZ#H~ej3|JY?0c3F0T2^OSa(<rGDQY$sCrnJ#e;xg=9U|@F^?#z;c
zB+%4oO~6QQHPLDU@h4V|K5TqbUre7gMqhlGRFmq1n)E^6C3?<GT>=_6nQ!i%d*<Bl
zobTN8!4r4K{pF^ndI5gP{o>qmcE_I}-|zHQF3`G#06am}vrULV>vlslyhkP@X9d-(
z`koh5AN4J?UUf}IX;_sOMW)fnZcMG)LvLKwQ)yS)6K)7^iMGwR(6;kuo~-dH@|=4@
zkfZ!(qR4JIc?=$-4L~KT2#@Pc*Se4w)`eZNClEnJZx3G;dqvsBKNM50oGkJ?Vu(ww
z9^&WsyOdxHhQKsLw@5@H?KgU=@4Mb0q=Aol7D*HT#M2S-!#=a&7_J%t;?+BzaYEv}
z&(}@T{JbxlIqYQxO=_6=rB!pjtmjuKGqkeaU%lb`-Ay9Zt;Rw8_=~34a2i8)0a%Ua
zE4rzb=o!Y!jCG<(QVfGKb6jWr=+*!rA7cZ&!M{D+gO&_>c}r!rQlt*W8c(amW>6?I
zn%7E8Wo8;Dt6%$1h>-eU&BxORj`br&%`gk)($Z-a;u#`9R9G9~)4^W8(sG+j@rgi>
z*Jj$!Hv?UBIXu=wZK99E+D^cx%+A4I*~bRZ!(XVVN?ut}w4yQt1n1G%sL~a!K=Zm%
zq6V8r6S>UnY-~yjSQr-CYURtpR&t4_gPY_oe>?b~i_8i9r{HLKY@I9+>=Gh8E$C^t
zVAie9@>8LUuFP_ZzY=;hP{1e%viRG#K3E(4*U-aVn4eRpXKSlkA=nJqHu$;lV{H^x
zYk`S*Wr?Orno&~B!hEW%r`Qz#C_F$S{B}5#o5B!t2pR%r#})ys21m9t%N3<)c?^ug
zinxoS+7)7U)trigO_o5^T%;p4i=Vp<ZH=-e^gSqx!;)`B;{G+*F}eXHQ4;T~ej53K
z03#k9Z0ZH0EHZqtUjAKl)a!8M(O8GqA<0K#aaudr={yfpbXA8wQp{4N$SU~x3c_K&
z_he-fZcPJNu7a{;6Sc_dzqx)E>m=LNA7ft<$pr74BAewm<Kz9d2rq&~CYEouSXRuT
z6U^4b*8Z>_AM;e=VgmzjhQ(pm_+}ys9r=1<5R(2X(Kq=LnD50oIm~LJSYCSBq`G=U
z94EVL80!Be@qi>xwI{+yba2{}Uun;fcdGBT!;HXNX?Zl0mJAl<9m%%V;~3q8@CbnH
zwr>N=@F$aUvz=WnMJvI-NhZkW)pwJB5GPA46{CEw^XfC$Oh?tQr7Piv-9mWX4svEW
z`1{X!woFIZI|uXibcWpGd+7xji9xPs9@=QeI5^qZ5W*z{H-a6dR#di(Hf%ul0>aA(
z*AZStu)`mKJucz$31~7p03^XpM6sTTb@fCFNK$ha-W^iVE450oMu|U=J(cVQKj@4)
z7@9L>Lo-+9xFyY0w4$l$1{>pB*&**J_^>#?mF*-k{!4a%$GekfUqd6-$C56<@EucN
ucH-&w@U+||MbrO_p(Twt1N1ZjXCuR}bcX|eAQ=#a4JlVoLZq3v_x}c*FNO*L

diff --git a/tests/test_environment.py b/tests/test_environment.py
index 2036aa4..eb94584 100644
--- a/tests/test_environment.py
+++ b/tests/test_environment.py
@@ -7,6 +7,7 @@
 from unittest import skipIf
 from packaging import version  # Fix: import version directly
 import torch
+import os  # Added for environment variable checks
 
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
@@ -20,8 +21,10 @@ def setUpClass(cls):
         cls.required_packages = {
             'torch': '2.0.1',
             'torchvision': '0.15.2',
-            'torchaudio': '2.0.1'
+            'torchaudio': '2.0.1',
+            'numpy': '1.21.0'  # Added numpy as an example
         }
+        cls.required_env_vars = ['HOME', 'PATH']  # Example environment variables
         cls._check_package_installations()
 
     @classmethod
@@ -124,7 +127,7 @@ def test_memory_allocation(self):
         import torch
         
         test_sizes = [(1000, 1000), (2000, 2000)]
-        devices = ['cpu'] + [f'cuda:{i}' for i in range(torch.cuda.device_count())] if torch.cuda.is_available() else []
+        devices = ['cpu'] + [f'cuda:{i}' for i in range(torch.cuda.device_count())] if torch.cuda.is_available() else ['cpu']
         
         for device in devices:
             for size in test_sizes:
@@ -216,6 +219,13 @@ def test_error_handling_and_logging(self):
             logger.error(f"Unexpected error during tensor creation: {str(e)}")
             self.fail(f"Unexpected error during tensor creation: {str(e)}")
 
+    def test_environment_variables(self):
+        """Verify essential environment variables are set"""
+        for var in self.required_env_vars:
+            value = os.getenv(var)
+            self.assertIsNotNone(value, f"Environment variable '{var}' is not set")
+            logger.info(f"Environment variable '{var}': {value}")
+
 if __name__ == '__main__':
     logger.info("Starting environment tests")
     logger.info(f"Platform: {platform.platform()}")