def test_that_gets_None_for_empty_mask_files(self): # Arrange command_iface.Clean() command_iface.SANS2D() # Act result = command_iface.GetTransmissionMask() # Assert self.assertEqual(None, result, 'The transmission mask should be None')
def test_that_gets_non_empty_mask_files(self): # Arrange trans_mask_files = ['mask_file1.xml', 'mask_file2.xml'] command_iface.Clean() command_iface.SANS2D() ReductionSingleton().transmission_calculator.mask_files = trans_mask_files # Act result = command_iface.GetTransmissionMask() # Assert self.assertEqual(trans_mask_files, result, 'The transmission mask should have two entries')