def test_zaxis_should_be_waited_for_buy_outputing_laser_off_signal(self, mock_LayerGenerator,mock_AudioWriter,mock_PathToAudio,mock_ZAxis,mock_LaserControl): mock_laser_control = mock_LaserControl.return_value mock_path_to_audio = mock_PathToAudio.return_value mock_zaxis = mock_ZAxis.return_value mock_audio_writer = mock_AudioWriter.return_value zaxis_return_values = [ 0.0,0.0, 0.25, 0.75, 1.0,1.0 ] def z_axis_side_effect(): return zaxis_return_values.pop(0) mock_zaxis.current_z_location_mm = z_axis_side_effect test_layer1 = Layer(0.0, [ LateralDraw([0.0,0.0],[2.0,2.0],2.0) ]) test_layer2 = Layer(1.0, [ LateralDraw([2.0,2.0],[0.0,0.0],2.0) ]) stub_layer_generator = StubLayerGenerator([test_layer1,test_layer2]) mock_path_to_audio.process.return_value = "SomeAudio" mock_laser_control.modulate.return_value = "SomeModulatedAudio" self.controller = Controller(mock_laser_control,mock_path_to_audio,mock_audio_writer,stub_layer_generator, mock_zaxis) self.controller.start() self.wait_for_controller() mock_zaxis.start.assert_called_with() self.assertEqual(4, mock_path_to_audio.process.call_count) self.assertEqual(2, mock_laser_control.set_laser_off.call_count) self.assertEqual(([2.0,2.0,0.0],[2.0,2.0,0.0],2.0), mock_path_to_audio.process.call_args_list[1][0]) self.assertEqual(([2.0,2.0,0.0],[2.0,2.0,0.0],2.0), mock_path_to_audio.process.call_args_list[2][0])
def test_should_work_with_no_writer(self, mock_LayerGenerator,mock_AudioWriter,mock_PathToAudio,mock_ZAxis,mock_LaserControl): mock_laser_control = mock_LaserControl.return_value mock_path_to_audio = mock_PathToAudio.return_value test_layer = Layer(0.0,[ LateralDraw([0.0,0.0],[2.0,2.0],100.0) ]) stub_layer_generator = StubLayerGenerator([test_layer]) mock_path_to_audio.process.return_value = "SomeAudio" mock_laser_control.modulate.return_value = "SomeModulatedAudio" self.controller = Controller(mock_laser_control,mock_path_to_audio,None,stub_layer_generator) self.controller.start() self.wait_for_controller()
def test_stop_should_stop_working_on_commands_when_generator_changed(self, mock_LayerGenerator,mock_AudioWriter,mock_PathToAudio,mock_ZAxis,mock_LaserControl): mock_laser_control = mock_LaserControl.return_value mock_path_to_audio = mock_PathToAudio.return_value mock_audio_writer = mock_AudioWriter.return_value mock_layer_generator = mock_LayerGenerator.return_value test_layer1 = Layer(1.0, [ LateralDraw([2.0,2.0],[0.0,0.0],2.0) for x in range(0,32768)]) stub_layer_generator1 = StubLayerGenerator([test_layer1]) test_layer2 = Layer(1.0, [ LateralDraw([0.0,0.0],[0.0,0.0],2.0)] ) stub_layer_generator2 = StubLayerGenerator([test_layer2]) mock_path_to_audio.process.return_value = "SomeAudio" mock_laser_control.modulate.return_value = "SomeModulatedAudio" self.controller = Controller(mock_laser_control,mock_path_to_audio,mock_audio_writer,stub_layer_generator1) self.controller.start() time.sleep(0.1) self.controller.change_generator(stub_layer_generator2) time.sleep(0.1) mock_path_to_audio.process.assert_called_with([0.0,0.0,1.0],[0.0,0.0,1.0],2.0) self.wait_for_controller()
def test_should_turn_on_laser_for_draw_commands(self, mock_LayerGenerator,mock_AudioWriter,mock_PathToAudio,mock_ZAxis,mock_LaserControl): mock_laser_control = mock_LaserControl.return_value mock_path_to_audio = mock_PathToAudio.return_value mock_audio_writer = mock_AudioWriter.return_value test_layer = Layer(0.0,[ LateralDraw([0.0,0.0],[2.0,2.0],100.0) ]) stub_layer_generator = StubLayerGenerator([test_layer]) mock_path_to_audio.process.return_value = "SomeAudio" mock_laser_control.modulate.return_value = "SomeModulatedAudio" self.controller = Controller(mock_laser_control,mock_path_to_audio,mock_audio_writer,stub_layer_generator) self.controller.start() self.wait_for_controller() self.assertEqual(1,mock_laser_control.set_laser_on.call_count)
def test_should_remember_current_posisition(self, mock_LayerGenerator,mock_AudioWriter,mock_PathToAudio,mock_ZAxis,mock_LaserControl): mock_laser_control = mock_LaserControl.return_value mock_path_to_audio = mock_PathToAudio.return_value mock_audio_writer = mock_AudioWriter.return_value test_layer = Layer(0.0,[ LateralDraw([0.0,0.0],[2.0,2.0],2.0), LateralDraw([2.0,2.0],[-1.0,-1.0],2.0) ]) stub_layer_generator = StubLayerGenerator([test_layer]) mock_path_to_audio.process.return_value = "SomeAudio" mock_laser_control.modulate.return_value = "SomeModulatedAudio" self.controller = Controller(mock_laser_control,mock_path_to_audio,mock_audio_writer,stub_layer_generator) self.controller.start() self.wait_for_controller() mock_path_to_audio.process.assert_called_with([2.0,2.0,0.0],[-1.0,-1.0,0.0],2.0)
def test_should_output_modulate_audio_for_movement_commands(self, mock_LayerGenerator,mock_AudioWriter,mock_PathToAudio,mock_ZAxis,mock_LaserControl): mock_laser_control = mock_LaserControl.return_value mock_path_to_audio = mock_PathToAudio.return_value mock_audio_writer = mock_AudioWriter.return_value test_layer = Layer(0.0,[ LateralDraw([0.0,0.0],[2.0,2.0],2.0) ]) stub_layer_generator = StubLayerGenerator([test_layer]) mock_path_to_audio.process.return_value = "SomeAudio" mock_laser_control.modulate.return_value = "SomeModulatedAudio" self.controller = Controller(mock_laser_control,mock_path_to_audio,mock_audio_writer,stub_layer_generator) self.controller.start() self.wait_for_controller() mock_laser_control.modulate.assert_called_with("SomeAudio") mock_audio_writer.write_chunk.assert_called_with("SomeModulatedAudio")
def test_should_record_errors_and_continue_when_abort_on_error_is_false(self, mock_LayerGenerator,mock_AudioWriter,mock_PathToAudio,mock_ZAxis,mock_LaserControl): mock_laser_control = mock_LaserControl.return_value mock_path_to_audio = mock_PathToAudio.return_value mock_audio_writer = mock_AudioWriter.return_value test_layer1 = Layer(0.0,[ LateralDraw([0.0,0.0],[2.0,2.0],100.0) ]) test_layer2 = Layer(0.0,[ LateralDraw([0.0,0.0],[2.0,2.0],100.0) ]) stub_layer_generator = StubLayerGenerator([test_layer1, test_layer2]) mock_path_to_audio.process.side_effect = Exception("Something Broke") mock_laser_control.modulate.return_value = "SomeModulatedAudio" self.controller = Controller(mock_laser_control,mock_path_to_audio,mock_audio_writer,stub_layer_generator, abort_on_error = False) self.controller.start() self.wait_for_controller() self.assertEquals(2, len(self.controller.get_status()['errors']))
def test_should_update_machine_status(self, mock_LayerGenerator,mock_AudioWriter,mock_PathToAudio,mock_ZAxis,mock_LaserControl): mock_laser_control = mock_LaserControl.return_value mock_path_to_audio = mock_PathToAudio.return_value mock_audio_writer = mock_AudioWriter.return_value test_layer = Layer(0.0,[ LateralDraw([0.0,0.0],[2.0,2.0],100.0) ]) stub_layer_generator = StubLayerGenerator([test_layer]) mock_path_to_audio.process.return_value = "SomeAudio" mock_laser_control.modulate.return_value = "SomeModulatedAudio" self.controller = Controller(mock_laser_control,mock_path_to_audio,mock_audio_writer,stub_layer_generator) self.controller.start() self.wait_for_controller() self.assertEquals(1, self.controller.get_status()['current_layer']) self.assertEquals('Complete',self.controller.get_status()['status'])
def test_should_ignore_z_in_layer_if_z_axis_none(self, mock_LayerGenerator,mock_AudioWriter,mock_PathToAudio,mock_ZAxis,mock_LaserControl): mock_laser_control = mock_LaserControl.return_value mock_path_to_audio = mock_PathToAudio.return_value mock_audio_writer = mock_AudioWriter.return_value test_layer1 = Layer(0.0, [ LateralDraw([0.0,0.0],[2.0,2.0],2.0) ]) test_layer2 = Layer(1.0, [ LateralDraw([2.0,2.0],[0.0,0.0],2.0) ]) stub_layer_generator = StubLayerGenerator([test_layer1,test_layer2]) mock_path_to_audio.process.return_value = "SomeAudio" mock_laser_control.modulate.return_value = "SomeModulatedAudio" self.controller = Controller(mock_laser_control,mock_path_to_audio,mock_audio_writer,stub_layer_generator,None) self.controller.start() self.wait_for_controller() self.assertEqual(2,mock_path_to_audio.process.call_count) mock_path_to_audio.process.assert_called_with([2.0,2.0,0.0],[0.0,0.0,1.0],2.0)
def test_if_move_command_start_and_current_pos_are_not_the_same_should_move_to_new_posisition(self, mock_LayerGenerator,mock_AudioWriter,mock_PathToAudio,mock_ZAxis,mock_LaserControl): mock_laser_control = mock_LaserControl.return_value mock_path_to_audio = mock_PathToAudio.return_value mock_audio_writer = mock_AudioWriter.return_value test_layer = Layer(0.0,[ LateralMove([0.0,0.0],[0.0,0.0],2.0), LateralMove([2.0,2.0],[-1.0,-1.0],2.0) ]) stub_layer_generator = StubLayerGenerator([test_layer]) mock_path_to_audio.process.return_value = "SomeAudio" mock_laser_control.modulate.return_value = "SomeModulatedAudio" self.controller = Controller(mock_laser_control,mock_path_to_audio,mock_audio_writer,stub_layer_generator) self.controller.start() self.wait_for_controller() mock_laser_control.modulate.call_count = 2 mock_path_to_audio.process.call_count = 2 mock_laser_control.set_laser_off.assert_called_with() self.assertEqual(([0.0,0.0,0.0],[-1.0,-1.0,0.0],2.0), mock_path_to_audio.process.call_args_list[1][0])
def test_stop_should_close_all_processes_cleanly_while_working_on_commands(self, mock_LayerGenerator,mock_AudioWriter,mock_PathToAudio,mock_ZAxis,mock_LaserControl): mock_laser_control = mock_LaserControl.return_value mock_path_to_audio = mock_PathToAudio.return_value mock_audio_writer = mock_AudioWriter.return_value mock_layer_generator = mock_LayerGenerator.return_value test_layer = Layer(1.0, [ LateralDraw([2.0,2.0],[0.0,0.0],2.0) for x in range(0,32768)]) stub_layer_generator = StubLayerGenerator([test_layer]) mock_path_to_audio.process.return_value = "SomeAudio" mock_laser_control.modulate.return_value = "SomeModulatedAudio" self.controller = Controller(mock_laser_control,mock_path_to_audio,mock_audio_writer,stub_layer_generator) self.controller.start() time.sleep(0.1) self.controller.stop() time.sleep(0.1) mock_audio_writer.close.assert_called_with() self.wait_for_controller()
def test_if_max_lead_specifed_should_skip_layers_if_higher(self, mock_LayerGenerator,mock_AudioWriter,mock_PathToAudio,mock_ZAxis,mock_LaserControl): max_lead_distance = 0.1 mock_laser_control = mock_LaserControl.return_value mock_path_to_audio = mock_PathToAudio.return_value mock_audio_writer = mock_AudioWriter.return_value mock_zaxis = mock_ZAxis.return_value zaxis_return_values = [ 0.0, 0.0, 2.0, 2.0,2.0,2.0 ] def z_axis_side_effect(): return zaxis_return_values.pop(0) mock_zaxis.current_z_location_mm = z_axis_side_effect test_layer1 = Layer(0.0,[ LateralDraw([0.0,0.0],[2.0,2.0],2.0), LateralDraw([2.0,2.0],[-1.0,-1.0],2.0) ]) test_layer2 = Layer(1.0,[ LateralDraw([0.0,0.0],[2.0,2.0],2.0), LateralDraw([2.0,2.0],[-1.0,-1.0],2.0) ]) test_layer3 = Layer(2.0,[ LateralDraw([0.0,0.0],[2.0,2.0],2.0), LateralDraw([2.0,2.0],[-1.0,-1.0],2.0) ]) stub_layer_generator = StubLayerGenerator([test_layer1, test_layer2, test_layer3]) mock_path_to_audio.process.return_value = "SomeAudio" mock_laser_control.modulate.return_value = "SomeModulatedAudio" self.controller = Controller(mock_laser_control,mock_path_to_audio,mock_audio_writer,stub_layer_generator,zaxis =mock_zaxis,max_lead_distance=max_lead_distance) self.controller.start() self.wait_for_controller() self.assertEquals(1, self.controller.get_status()['skipped_layers']) self.assertEquals(5, mock_audio_writer.write_chunk.call_count)