def test_stage_write_erase(self): self.device.execute = Mock(return_value=self.raw_output.write_erase) # Execute stage: write_erase with self.assertRaises(AEtestFailedSignal): write_erase(self.section, self.steps, self.device, **self.device.clean.write_erase)
def test_stage_write_erase(self): self.device.execute = Mock(return_value=self.raw_output.write_erase, error_pattern=['^%\\s*[Ii]nvalid (command|input)']) # Execute stage: write_erase with self.assertRaises(TerminateStepSignal): write_erase(self.section, self.steps, self.device, **self.device.clean.write_erase)