@@ -448,7 +448,7 @@ def tearDown(self):
448448 self .view .deleteLater ()
449449
450450 def test_single_file_yaml_backward_compatibility (self ):
451- """Test that single-file YAML loading still works (backward compatibility) ."""
451+ """Test that single-file YAML loading still works."""
452452 with tempfile .TemporaryDirectory () as temp_dir :
453453 temp_path = Path (temp_dir )
454454
@@ -501,6 +501,7 @@ def test_single_file_yaml_backward_compatibility(self):
501501
502502 def test_multi_file_yaml_loading (self ):
503503 """Test loading YAML with multiple files per category."""
504+ contr = self .controller .measurement_controller
504505 with tempfile .TemporaryDirectory () as temp_dir :
505506 temp_path = Path (temp_dir )
506507
@@ -558,21 +559,15 @@ def test_multi_file_yaml_loading(self):
558559 # Call the method
559560 self .controller .open_yaml_and_load_files (str (yaml_file ))
560561
561- # Verify measurement files were loaded (once in overwrite, once in append)
562- self .assertEqual (
563- self .controller .measurement_controller .open_table .call_count , 2
564- )
562+ # Verify measurement files loading (once in overwrite/append)
563+ self .assertEqual (contr .open_table .call_count , 2 )
565564
566565 # Check that first call was with mode='overwrite'
567- first_call = self .controller .measurement_controller .open_table .call_args_list [
568- 0
569- ]
566+ first_call = contr .open_table .call_args_list [0 ]
570567 self .assertEqual (first_call [1 ].get ("mode" ), "overwrite" )
571568
572569 # Check that second call was with mode='append'
573- second_call = self .controller .measurement_controller .open_table .call_args_list [
574- 1
575- ]
570+ second_call = contr .open_table .call_args_list [1 ]
576571 self .assertEqual (second_call [1 ].get ("mode" ), "append" )
577572
578573 # Verify observable files were loaded
0 commit comments