def setUp(self):
options = {
'limit_switches': True,
'firmware': 'v1.0.5',
'config': {
'ot_version': 'one_pro',
'version': 'v1.2.0', # config version
'alpha_steps_per_mm': 80.0,
'beta_steps_per_mm': 80.0
}
}
self.s = VirtualSmoothie(port=None, options=options)
self.s.open()
def setUp(self):
options = {
"limit_switches": True,
"firmware": "v1.0.5",
"config": {
"ot_version": "one_pro",
"version": "v1.2.0", # config version
"alpha_steps_per_mm": 80.0,
"beta_steps_per_mm": 80.0,
},
}
self.s = VirtualSmoothie(port=None, options=options)
self.s.open()
def get_virtual_device(self, port=None, options=None):
"""
Connect to a :class:`VirtualSmoothie` to simulate behavior of
a Smoothieboard
Parameters
----------
port : str
Port name. Could be `None` or anything.
options : dict
Options to be passed to :class:`VirtualSmoothie`.
Default:
::
default_options = {
'limit_switches': True,
'firmware': 'v1.0.5',
'config': {
'ot_version': 'one_pro',
'version': 'v1.0.3', # config version
'alpha_steps_per_mm': 80.0,
'beta_steps_per_mm': 80.0,
'gamma_steps_per_mm': 1068.7
}
}
"""
default_options = {
'limit_switches': True,
'firmware': 'v1.0.5',
'config': {
'ot_version': 'one_pro',
'version': 'v1.2.0', # config version
'alpha_steps_per_mm': 80.0,
'beta_steps_per_mm': 80.0,
'gamma_steps_per_mm': 1068.7
}
}
if options:
default_options['config'].update(options.get('config', {}))
options['config'] = default_options['config']
default_options.update(options)
return VirtualSmoothie(port=port, options=default_options)
class VirtualSmoothieTestCase(unittest.TestCase):
def setUp(self):
options = {
"limit_switches": True,
"firmware": "v1.0.5",
"config": {
"ot_version": "one_pro",
"version": "v1.2.0", # config version
"alpha_steps_per_mm": 80.0,
"beta_steps_per_mm": 80.0,
},
}
self.s = VirtualSmoothie(port=None, options=options)
self.s.open()
def test_version(self):
self.s.write("version")
res = self.s.readline()
self.assertEqual(res, b'{"version":v1.0.5}')
def test_reset(self):
self.s.write("reset")
res = self.s.readline()
self.assertEqual(res, b"Smoothie out. Peace. Rebooting in 5 seconds...")
def test_config_get(self):
self.s.write("config-get sd ot_version")
res = self.s.readline()
self.assertEqual(res, b"sd: ot_version is set to one_pro")
self.s.write("config-get sd go_crazy")
res = self.s.readline()
self.assertEqual(res, b"sd: go_crazy is not in config")
def test_config_set(self):
self.s.write("config-set sd ot_version hood")
res = self.s.readline()
self.assertEqual(res, b"sd: ot_version has been set to hood")
self.s.write("config-get sd ot_version")
res = self.s.readline()
self.assertEqual(res, b"sd: ot_version is set to hood")
def test_parse_command(self):
expected_result = {"command": "G0", "arguments": {"X": 1.0, "Y": 2.5, "Z": 3.0, "a": 4.0, "b": 5.0, "F": -66.0}}
res = self.s.parse_command("G0X1Y2.5Z3a4b5F-66")
self.assertDictEqual(res, expected_result)
res = self.s.parse_command("G0 X1Y2.5Z3a4b5F-66")
self.assertDictEqual(res, expected_result)
res = self.s.parse_command("G0X1 Y2.5Z3a4b5F-66")
self.assertDictEqual(res, expected_result)
def test_set_position(self):
self.s.write("G92X1Y2.5Z3")
response = self.s.readline()
self.assertEqual(response, b"ok")
self.s.write("M114")
response = self.s.readline()
self.assertEqual(response[:3], b"ok ")
response = json.loads(response[3:].decode("utf-8"))
expected_result = {
"M114": {"X": 1.0, "Y": 2.5, "Z": 3.0, "A": 0.0, "B": 0.0, "x": 1.0, "y": 2.5, "z": 3.0, "a": 0.0, "b": 0.0}
}
self.assertDictEqual(response, expected_result)
def test_endstops(self):
self.s.write("M119")
response = self.s.readline()
response = json.loads(response.decode("utf-8"))
expected_result = {"M119": {"min_x": 0, "min_y": 0, "min_z": 0, "min_a": 0, "min_b": 0}}
self.assertDictEqual(response, expected_result)
response = self.s.readline()
self.assertEqual(response, b"ok")
def test_calm_down(self):
self.s.write("M999")
response = self.s.readline()
self.assertEqual(response, b"ok")
def test_dwell(self):
self.s.write("G4 S1 P200")
response = self.s.readline()
self.assertEqual(response, b"ok")
def test_home(self):
self.s.write("G0X1Y2.5Z3")
response = self.s.readline()
self.assertEqual(response, b"ok")
self.s.write("G28X")
response = self.s.readline()
self.assertEqual(response, b"ok")
expected_result = {
"M114": {"X": 0.0, "Y": 2.5, "Z": 3.0, "A": 0.0, "B": 0.0, "x": 0.0, "y": 2.5, "z": 3.0, "a": 0.0, "b": 0.0}
}
self.s.write("M114")
response = self.s.readline()
self.assertEqual(response[:3], b"ok ")
response = json.loads(response[3:].decode("utf-8"))
self.assertDictEqual(response, expected_result)
self.s.write("G28")
response = self.s.readline()
self.assertEqual(response, b"ok")
expected_result = {
"M114": {"X": 0.0, "Y": 0.0, "Z": 0.0, "A": 0.0, "B": 0.0, "x": 0.0, "y": 0.0, "z": 0.0, "a": 0.0, "b": 0.0}
}
self.s.write("M114")
response = self.s.readline()
self.assertEqual(response[:3], b"ok ")
response = json.loads(response[3:].decode("utf-8"))
self.assertDictEqual(response, expected_result)
def test_limit_switch_hit(self):
self.s.write("G90")
response = self.s.readline()
self.assertEqual(response, b"ok")
self.s.write("G0X-100")
response = self.s.readline()
self.assertEqual(response, b"ok")
response = self.s.readline()
self.assertEqual(response, b'{"limit":"min_x"}')
def test_mosfet_state(self):
for i in range(12):
self.s.write("M{}".format(i + 40))
response = self.s.readline()
self.assertEqual(response, b"ok")
def test_power_on_off(self):
self.s.write("M17")
response = self.s.readline()
self.assertEqual(response, b"ok")
self.s.write("M18")
response = self.s.readline()
self.assertEqual(response, b"ok")
def test_move(self):
self.s.write("G90")
response = self.s.readline()
self.assertEqual(response, b"ok")
self.s.write("G0X1Y2.5Z3")
response = self.s.readline()
self.assertEqual(response, b"ok")
self.s.write("M114")
response = self.s.readline()
self.assertEqual(response[:3], b"ok ")
response = json.loads(response[3:].decode("utf-8"))
expected_result = {
"M114": {"X": 1.0, "Y": 2.5, "Z": 3.0, "A": 0.0, "B": 0.0, "x": 1.0, "y": 2.5, "z": 3.0, "a": 0.0, "b": 0.0}
}
self.assertDictEqual(response, expected_result)
self.s.write("G91")
response = self.s.readline()
self.assertEqual(response, b"ok")
self.s.write("G0X1Y1Z-1")
response = self.s.readline()
self.assertEqual(response, b"ok")
self.s.write("M114")
response = self.s.readline()
self.assertEqual(response[:3], b"ok ")
response = json.loads(response[3:].decode("utf-8"))
expected_result = {
"M114": {"X": 2.0, "Y": 3.5, "Z": 2.0, "A": 0.0, "B": 0.0, "x": 2.0, "y": 3.5, "z": 2.0, "a": 0.0, "b": 0.0}
}
self.assertDictEqual(response, expected_result)
class VirtualSmoothieTestCase(unittest.TestCase):
def setUp(self):
options = {
'limit_switches': True,
'firmware': 'v1.0.5',
'config': {
'ot_version': 'one_pro',
'version': 'v1.2.0', # config version
'alpha_steps_per_mm': 80.0,
'beta_steps_per_mm': 80.0
}
}
self.s = VirtualSmoothie(port=None, options=options)
self.s.open()
def test_version(self):
self.s.write('version')
res = self.s.readline()
self.assertEqual(res, b'{"version":v1.0.5}')
def test_reset(self):
self.s.write('reset')
res = self.s.readline()
self.assertEqual(res,
b'Smoothie out. Peace. Rebooting in 5 seconds...')
def test_config_get(self):
self.s.write('config-get sd ot_version')
res = self.s.readline()
self.assertEqual(res, b'sd: ot_version is set to one_pro')
self.s.write('config-get sd go_crazy')
res = self.s.readline()
self.assertEqual(res, b'sd: go_crazy is not in config')
def test_config_set(self):
self.s.write('config-set sd ot_version hood')
res = self.s.readline()
self.assertEqual(res, b'sd: ot_version has been set to hood')
self.s.write('config-get sd ot_version')
res = self.s.readline()
self.assertEqual(res, b'sd: ot_version is set to hood')
def test_parse_command(self):
expected_result = {
'command': 'G0',
'arguments': {
'X': 1.0,
'Y': 2.5,
'Z': 3.0,
'a': 4.0,
'b': 5.0,
'F': -66.0
}
}
res = self.s.parse_command('G0X1Y2.5Z3a4b5F-66')
self.assertDictEqual(res, expected_result)
res = self.s.parse_command('G0 X1Y2.5Z3a4b5F-66')
self.assertDictEqual(res, expected_result)
res = self.s.parse_command('G0X1 Y2.5Z3a4b5F-66')
self.assertDictEqual(res, expected_result)
def test_set_position(self):
self.s.write('G92X1Y2.5Z3')
response = self.s.readline()
self.assertEqual(response, b'ok')
self.s.write('M114')
response = self.s.readline()
self.assertEqual(response[:3], b'ok ')
response = json.loads(response[3:].decode('utf-8'))
expected_result = {
'M114': {
'X': 1.0,
'Y': 2.5,
'Z': 3.0,
'A': 0.0,
'B': 0.0,
'x': 1.0,
'y': 2.5,
'z': 3.0,
'a': 0.0,
'b': 0.0
}
}
self.assertDictEqual(response, expected_result)
def test_endstops(self):
self.s.write('M119')
response = self.s.readline()
response = json.loads(response.decode('utf-8'))
expected_result = {
"M119": {
"min_x": 0,
"min_y": 0,
"min_z": 0,
"min_a": 0,
"min_b": 0
}
}
self.assertDictEqual(response, expected_result)
response = self.s.readline()
self.assertEqual(response, b'ok')
def test_calm_down(self):
self.s.write('M999')
response = self.s.readline()
self.assertEqual(response, b'ok')
def test_dwell(self):
self.s.write('G4 S1 P200')
response = self.s.readline()
self.assertEqual(response, b'ok')
def test_home(self):
self.s.write('G0X1Y2.5Z3')
response = self.s.readline()
self.assertEqual(response, b'ok')
self.s.write('G28X')
response = self.s.readline()
self.assertEqual(response, b'ok')
expected_result = {
'M114': {
'X': 0.0,
'Y': 2.5,
'Z': 3.0,
'A': 0.0,
'B': 0.0,
'x': 0.0,
'y': 2.5,
'z': 3.0,
'a': 0.0,
'b': 0.0
}
}
self.s.write('M114')
response = self.s.readline()
self.assertEqual(response[:3], b'ok ')
response = json.loads(response[3:].decode('utf-8'))
self.assertDictEqual(response, expected_result)
self.s.write('G28')
response = self.s.readline()
self.assertEqual(response, b'ok')
expected_result = {
'M114': {
'X': 0.0,
'Y': 0.0,
'Z': 0.0,
'A': 0.0,
'B': 0.0,
'x': 0.0,
'y': 0.0,
'z': 0.0,
'a': 0.0,
'b': 0.0
}
}
self.s.write('M114')
response = self.s.readline()
self.assertEqual(response[:3], b'ok ')
response = json.loads(response[3:].decode('utf-8'))
self.assertDictEqual(response, expected_result)
def test_limit_switch_hit(self):
self.s.write('G90')
response = self.s.readline()
self.assertEqual(response, b'ok')
self.s.write('G0X-100')
response = self.s.readline()
self.assertEqual(response, b'ok')
response = self.s.readline()
self.assertEqual(response, b'{"limit":"min_x"}')
def test_mosfet_state(self):
for i in range(12):
self.s.write('M{}'.format(i + 40))
response = self.s.readline()
self.assertEqual(response, b'ok')
def test_power_on_off(self):
self.s.write('M17')
response = self.s.readline()
self.assertEqual(response, b'ok')
self.s.write('M18')
response = self.s.readline()
self.assertEqual(response, b'ok')
def test_move(self):
self.s.write('G90')
response = self.s.readline()
self.assertEqual(response, b'ok')
self.s.write('G0X1Y2.5Z3')
response = self.s.readline()
self.assertEqual(response, b'ok')
self.s.write('M114')
response = self.s.readline()
self.assertEqual(response[:3], b'ok ')
response = json.loads(response[3:].decode('utf-8'))
expected_result = {
'M114': {
'X': 1.0,
'Y': 2.5,
'Z': 3.0,
'A': 0.0,
'B': 0.0,
'x': 1.0,
'y': 2.5,
'z': 3.0,
'a': 0.0,
'b': 0.0
}
}
self.assertDictEqual(response, expected_result)
self.s.write('G91')
response = self.s.readline()
self.assertEqual(response, b'ok')
self.s.write('G0X1Y1Z-1')
response = self.s.readline()
self.assertEqual(response, b'ok')
self.s.write('M114')
response = self.s.readline()
self.assertEqual(response[:3], b'ok ')
response = json.loads(response[3:].decode('utf-8'))
expected_result = {
'M114': {
'X': 2.0,
'Y': 3.5,
'Z': 2.0,
'A': 0.0,
'B': 0.0,
'x': 2.0,
'y': 3.5,
'z': 2.0,
'a': 0.0,
'b': 0.0
}
}
self.assertDictEqual(response, expected_result)