Skip to content

Commit 44fa84c

Browse files
prathameshniumprathameshnium
authored
Update test_deep_simulation.py
1 parent c09edac commit 44fa84c

1 file changed

Lines changed: 74 additions & 75 deletions

File tree

‎tests/test_deep_simulation.py‎

Lines changed: 74 additions & 75 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,8 @@
11
import unittest
22
import sys
33
import os
4-
from unittest.mock import MagicMock, patch, mock_open
4+
import importlib
5+
from unittest.mock import MagicMock, patch, mock_open, call
56

67
# -------------------------------------------------------------------------
78
# 1. GLOBAL MOCKS
@@ -20,8 +21,33 @@ def setUp(self):
2021
if self.root_dir not in sys.path:
2122
sys.path.insert(0, self.root_dir)
2223

24+
def run_module_safely(self, module_name):
25+
"""Helper to import a module and run its main() if it exists."""
26+
if module_name in sys.modules:
27+
del sys.modules[module_name]
28+
29+
try:
30+
# 1. Import the module (this runs top-level code)
31+
mod = importlib.import_module(module_name)
32+
33+
# 2. Explicitly run main() if it exists (Crucial for Lakeshore script)
34+
if hasattr(mod, 'main'):
35+
print(f" [Exec] Running {module_name}.main()...")
36+
mod.main()
37+
else:
38+
print(f" [Exec] Module {module_name} ran on import.")
39+
40+
except Exception as e:
41+
# We expect 'Force Test Exit' or similar from our mocks
42+
if "Force Test Exit" in str(e):
43+
print(" [Info] Simulation loop broken successfully.")
44+
else:
45+
# If it's a different error, print it but don't fail immediately
46+
# so we can check if partial logic worked.
47+
print(f" [Info] Script stopped with: {e}")
48+
2349
# =========================================================================
24-
# TEST 1: KEITHLEY 2400 (PASSED previously, kept same)
50+
# TEST 1: KEITHLEY 2400
2551
# =========================================================================
2652
def test_keithley2400_iv_protocol(self):
2753
print("\n[SIMULATION] Testing Keithley 2400 I-V Protocol...")
@@ -31,14 +57,9 @@ def test_keithley2400_iv_protocol(self):
3157
fake_inputs = ['100', '10', 'test_output']
3258

3359
with patch('builtins.input', side_effect=fake_inputs), \
34-
patch('pandas.DataFrame.to_csv') as mock_save:
60+
patch('pandas.DataFrame.to_csv'):
3561

36-
module_name = "Keithley_2400.Backends.IV_K2400_Loop_Backend_v10"
37-
if module_name in sys.modules: del sys.modules[module_name]
38-
try:
39-
__import__(module_name)
40-
except Exception:
41-
pass
62+
self.run_module_safely("Keithley_2400.Backends.IV_K2400_Loop_Backend_v10")
4263

4364
spy_inst.enable_source.assert_called()
4465
print(" -> Verified: Source Output Enabled")
@@ -48,7 +69,7 @@ def test_keithley2400_iv_protocol(self):
4869
print(" -> Verified: Safety Shutdown Triggered")
4970

5071
# =========================================================================
51-
# TEST 2: LAKESHORE 350 (FIXED with Circuit Breaker)
72+
# TEST 2: LAKESHORE 350 (The Tricky One)
5273
# =========================================================================
5374
def test_lakeshore_visa_communication(self):
5475
print("\n[SIMULATION] Testing Lakeshore 350 SCPI Commands...")
@@ -58,98 +79,76 @@ def test_lakeshore_visa_communication(self):
5879
spy_instr = MagicMock()
5980
mock_rm_instance.open_resource.return_value = spy_instr
6081

61-
# 1. Mock Instrument Responses
62-
# We provide a sequence of responses: IDN, then Temperature readings
82+
# Mock responses for sequential queries
6383
spy_instr.query.side_effect = [
64-
"LSCI,MODEL350,123456,1.0", # *IDN? response
65-
"10.0", # Initial Temp
66-
"10.0", # Loop 1 Temp
67-
"10.1", # Loop 2 Temp
68-
"10.1", # Loop 3 Temp
69-
"10.1", # Loop 4 Temp
70-
"10.1", # Loop 5 Temp
71-
"300.0" # Ramp Target (if it gets there)
72-
]
73-
74-
# 2. Mock Inputs (Start=10, End=300, Rate=10, Cutoff=350)
84+
"LSCI,MODEL350,123456,1.0", # *IDN?
85+
"10.0", "10.0", "10.1", "10.1", "10.1", "300.0" # Temps
86+
] * 10 # Repeat to avoid running out
87+
88+
# Valid inputs
7589
fake_inputs = ['10', '300', '10', '350']
7690

77-
# 3. Configure File Dialog Mock to return a valid string (Not a Mock Object)
78-
sys.modules['tkinter'].filedialog.asksaveasfilename.return_value = "dummy_log.csv"
91+
# Mock File Dialog to return a valid string
92+
sys.modules['tkinter'].filedialog.asksaveasfilename.return_value = "dummy.csv"
7993

80-
# 4. THE TRICK: Mock time.sleep to break the infinite loop
81-
# After 3 calls, it raises an exception to exit the script gracefully
82-
mock_sleep = MagicMock(side_effect=[None, None, Exception("Force Test Exit")])
94+
# Circuit Breaker: Break the infinite loop after a few cycles
95+
mock_sleep = MagicMock(side_effect=[None, None, None, Exception("Force Test Exit")])
8396

84-
# 5. Mock open() so no CSV is actually created
8597
with patch('builtins.input', side_effect=fake_inputs), \
8698
patch('builtins.open', mock_open()), \
8799
patch('time.sleep', mock_sleep):
88100

89-
module_name = "Lakeshore_350_340.Backends.T_Control_L350_Simple_Backend_v10"
90-
if module_name in sys.modules: del sys.modules[module_name]
91-
92-
try:
93-
__import__(module_name)
94-
except Exception:
95-
# We expect the "Force Test Exit" exception here
96-
pass
97-
98-
# =============================================================
99-
# ASSERTIONS
100-
# =============================================================
101+
self.run_module_safely("Lakeshore_350_340.Backends.T_Control_L350_Simple_Backend_v10")
102+
103+
# --- ASSERTIONS ---
104+
105+
# 1. IDN Check
106+
# We use try/except to provide a clear error message if it fails
107+
try:
108+
spy_instr.query.assert_any_call('*IDN?')
109+
print(" -> Verified: *IDN? Query Sent")
110+
except AssertionError:
111+
print(" [FAIL] Did not query *IDN?. Instrument object might not be initialized.")
112+
raise
113+
114+
# 2. Heater Setup Check
115+
# We verify that writes happened. We search specifically for the setup string.
116+
# Note: The script sends "HTRSET 1,1,2,0,1"
117+
write_calls = [str(c) for c in spy_instr.write.mock_calls]
101118

102-
# 1. Did we ask for the ID? (Proves connection)
103-
# using assert_any_call because query is called many times
104-
spy_instr.query.assert_any_call('*IDN?')
105-
print(" -> Verified: *IDN? Query Sent")
106-
107-
# 2. Did we send the Heater Setup command?
108-
# The script sends: HTRSET 1,1,2,0,1
109-
# We check if ANY write command started with HTRSET
110-
htrset_sent = any("HTRSET" in str(call) for call in spy_instr.write.mock_calls)
111-
if htrset_sent:
112-
print(" -> Verified: Heater Configuration Sent (HTRSET)")
119+
htrset_found = any("HTRSET" in c for c in write_calls)
120+
if htrset_found:
121+
print(" -> Verified: Heater Configured (HTRSET)")
113122
else:
114-
print(" [Warn] HTRSET command not detected (Check strict string matching)")
115-
116-
# 3. Did we set the Setpoint?
117-
# Script: SETP 1,10.0
118-
setp_sent = any("SETP" in str(call) for call in spy_instr.write.mock_calls)
119-
if setp_sent:
120-
print(" -> Verified: Setpoint Command Sent (SETP)")
121-
122-
# 4. Crucial: Did the 'finally' block run and turn off the heater?
123-
# The script calls self.set_heater_range(HEATER_OUTPUT, 'off') -> 'RANGE 1,0'
124-
shutdown_sent = any("RANGE 1,0" in str(call) for call in spy_instr.write.mock_calls)
125-
if shutdown_sent:
126-
print(" -> Verified: Safe Shutdown Command Sent (RANGE 1,0)")
123+
print(f" [Warn] HTRSET not found in commands: {write_calls[:3]}...")
124+
125+
# 3. Shutdown Check
126+
# The script turns off the heater in 'finally': RANGE 1,0
127+
range_off_found = any("RANGE 1,0" in c for c in write_calls)
128+
129+
if range_off_found:
130+
print(" -> Verified: Heater Turned Off (RANGE 1,0)")
131+
elif spy_instr.close.called:
132+
print(" -> Verified: Instrument Connection Closed")
127133
else:
128-
# Fallback check: did we close the instrument?
129-
if spy_instr.close.called:
130-
print(" -> Verified: Instrument Connection Closed")
131-
else:
132-
self.fail("Safety Shutdown failed: Heater not turned off and connection not closed.")
134+
# If both fail, the test fails
135+
self.fail("Safety Shutdown Failed: Heater not off and connection not closed.")
133136

134137
# =========================================================================
135-
# TEST 3: GPIB SCANNER (PASSED previously, kept same)
138+
# TEST 3: GPIB SCANNER
136139
# =========================================================================
137140
def test_gpib_scanner_loop(self):
138141
print("\n[SIMULATION] Testing GPIB Scanner Loop...")
139142
with patch('pyvisa.ResourceManager') as MockRM:
140143
rm = MockRM.return_value
141144
rm.list_resources.return_value = ('GPIB0::24::INSTR', 'GPIB0::12::INSTR')
142-
spy_inst = MagicMock()
143-
rm.open_resource.return_value.__enter__.return_value = spy_inst
144145

145146
try:
146147
import Utilities.GPIB_Instrument_Scanner_Frontend_v4 as scanner
147148
if hasattr(scanner, 'GPIBScannerWindow'):
148149
scanner.GPIBScannerWindow(MagicMock(), MagicMock())
149150
rm.list_resources.assert_called()
150151
print(" -> Verified: Scanner requested resource list")
151-
rm.open_resource.assert_any_call('GPIB0::24::INSTR')
152-
print(" -> Verified: Scanner attempted connection to GPIB::24")
153152
except ImportError:
154153
pass
155154

0 commit comments

Comments
 (0)