File tree 1 file changed +8
-7
lines changed
1 file changed +8
-7
lines changed Original file line number Diff line number Diff line change @@ -2380,17 +2380,18 @@ def test_with_reinit_reloaded(self):
2380
2380
# Keep a reference around.
2381
2381
basic = self .load (self .NAME )
2382
2382
2383
- for name in [
2384
- f'{ self .NAME } _with_reinit' , # m_size == 0
2385
- f'{ self .NAME } _with_state' , # m_size > 0
2383
+ for name , has_state in [
2384
+ ( f'{ self .NAME } _with_reinit' , False ) , # m_size == 0
2385
+ ( f'{ self .NAME } _with_state' , True ), # m_size > 0
2386
2386
]:
2387
2387
self .add_module_cleanup (name )
2388
- with self .subTest (name ):
2388
+ with self .subTest (name = name , has_state = has_state ):
2389
2389
loaded = self .load (name )
2390
- reloaded = self .re_load (name , loaded .module )
2391
-
2392
- if name .endswith ("_with_state" ):
2390
+ if has_state :
2393
2391
self .addCleanup (loaded .module ._clear_module_state )
2392
+
2393
+ reloaded = self .re_load (name , loaded .module )
2394
+ if has_state :
2394
2395
self .addCleanup (reloaded .module ._clear_module_state )
2395
2396
2396
2397
self .check_common (loaded )
You can’t perform that action at this time.
0 commit comments