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)
