191 lines
7.9 KiB
Python
191 lines
7.9 KiB
Python
import sys
|
|
import unittest
|
|
from PyQt5 import QtGui, QtWidgets
|
|
from FlatCAMApp import App, tclCommands
|
|
from FlatCAMObj import FlatCAMGerber, FlatCAMGeometry, FlatCAMCNCjob
|
|
from flatcamGUI.ObjectUI import GerberObjectUI, GeometryObjectUI
|
|
from time import sleep
|
|
import os
|
|
import tempfile
|
|
|
|
|
|
class GerberFlowTestCase(unittest.TestCase):
|
|
"""
|
|
This is a top-level test covering the Gerber-to-GCode
|
|
generation workflow.
|
|
|
|
THIS IS A REQUIRED TEST FOR ANY UPDATES.
|
|
|
|
"""
|
|
|
|
filename = 'simple1.gbr'
|
|
|
|
def setUp(self):
|
|
self.app = QtWidgets.QApplication(sys.argv)
|
|
|
|
# Create App, keep app defaults (do not load
|
|
# user-defined defaults).
|
|
self.fc = App(user_defaults=False)
|
|
|
|
self.fc.open_gerber('tests/gerber_files/' + self.filename)
|
|
|
|
def tearDown(self):
|
|
del self.fc
|
|
del self.app
|
|
|
|
def test_flow(self):
|
|
# Names of available objects.
|
|
names = self.fc.collection.get_names()
|
|
print(names)
|
|
|
|
# --------------------------------------
|
|
# Total of 1 objects.
|
|
# --------------------------------------
|
|
self.assertEqual(len(names), 1,
|
|
"Expected 1 object, found %d" % len(names))
|
|
|
|
# --------------------------------------
|
|
# Object's name matches the file name.
|
|
# --------------------------------------
|
|
self.assertEqual(names[0], self.filename,
|
|
"Expected name == %s, got %s" % (self.filename, names[0]))
|
|
|
|
# ---------------------------------------
|
|
# Get object by that name, make sure it's a FlatCAMGerber.
|
|
# ---------------------------------------
|
|
gerber_name = names[0]
|
|
gerber_obj = self.fc.collection.get_by_name(gerber_name)
|
|
self.assertTrue(isinstance(gerber_obj, FlatCAMGerber),
|
|
"Expected FlatCAMGerber, instead, %s is %s" %
|
|
(gerber_name, type(gerber_obj)))
|
|
|
|
# ----------------------------------------
|
|
# Object's GUI matches Object's options
|
|
# ----------------------------------------
|
|
# TODO: Open GUI with double-click on object.
|
|
# Opens the Object's GUI, populates it.
|
|
gerber_obj.build_ui()
|
|
for option, value in list(gerber_obj.options.items()):
|
|
try:
|
|
form_field = gerber_obj.form_fields[option]
|
|
except KeyError:
|
|
print(("**********************************************************\n"
|
|
"* WARNING: Option '{}' has no form field\n"
|
|
"**********************************************************"
|
|
"".format(option)))
|
|
continue
|
|
self.assertEqual(value, form_field.get_value(),
|
|
"Option '{}' == {} but form has {}".format(
|
|
option, value, form_field.get_value()
|
|
))
|
|
|
|
# --------------------------------------------------
|
|
# Changes in the GUI should be read in when
|
|
# running any process. Changing something here.
|
|
# --------------------------------------------------
|
|
|
|
form_field = gerber_obj.form_fields['isotooldia']
|
|
value = form_field.get_value()
|
|
form_field.set_value(value * 1.1) # Increase by 10%
|
|
print(("'isotooldia' == {}".format(value)))
|
|
|
|
# --------------------------------------------------
|
|
# Create isolation routing using default values
|
|
# and by clicking on the button.
|
|
# --------------------------------------------------
|
|
# Get the object's GUI and click on "Generate Geometry" under
|
|
# "Isolation Routing"
|
|
assert isinstance(gerber_obj, FlatCAMGerber) # Just for the IDE
|
|
# Changed: UI has been build already
|
|
#gerber_obj.build_ui() # Open the object's UI.
|
|
ui = gerber_obj.ui
|
|
assert isinstance(ui, GerberObjectUI)
|
|
ui.generate_iso_button.click() # Click
|
|
|
|
# ---------------------------------------------
|
|
# Check that GUI has been read in.
|
|
# ---------------------------------------------
|
|
value = gerber_obj.options['isotooldia']
|
|
form_value = form_field.get_value()
|
|
self.assertEqual(value, form_value,
|
|
"Form value for '{}' == {} was not read into options"
|
|
"which has {}".format('isotooldia', form_value, value))
|
|
print(("'isotooldia' == {}".format(value)))
|
|
|
|
# ---------------------------------------------
|
|
# Check that only 1 object has been created.
|
|
# ---------------------------------------------
|
|
names = self.fc.collection.get_names()
|
|
self.assertEqual(len(names), 2,
|
|
"Expected 2 objects, found %d" % len(names))
|
|
|
|
# -------------------------------------------------------
|
|
# Make sure the Geometry Object has the correct name
|
|
# -------------------------------------------------------
|
|
geo_name = gerber_name + "_iso"
|
|
self.assertTrue(geo_name in names,
|
|
"Object named %s not found." % geo_name)
|
|
|
|
# -------------------------------------------------------
|
|
# Get the object make sure it's a geometry object
|
|
# -------------------------------------------------------
|
|
geo_obj = self.fc.collection.get_by_name(geo_name)
|
|
self.assertTrue(isinstance(geo_obj, FlatCAMGeometry),
|
|
"Expected a FlatCAMGeometry, got %s" % type(geo_obj))
|
|
|
|
# ------------------------------------
|
|
# Open the UI, make CNCObject
|
|
# ------------------------------------
|
|
geo_obj.build_ui()
|
|
ui = geo_obj.ui
|
|
assert isinstance(ui, GeometryObjectUI) # Just for the IDE
|
|
ui.generate_cnc_button.click() # Click
|
|
|
|
# Work is done in a separate thread and results are
|
|
# passed via events to the main event loop which is
|
|
# not running. Run only for pending events.
|
|
#
|
|
# I'm not sure why, but running it only once does
|
|
# not catch the new object. Might be a timing issue.
|
|
# http://pyqt.sourceforge.net/Docs/PyQt4/qeventloop.html#details
|
|
for _ in range(2):
|
|
sleep(0.1)
|
|
self.app.processEvents()
|
|
|
|
# ---------------------------------------------
|
|
# Check that only 1 object has been created.
|
|
# ---------------------------------------------
|
|
names = self.fc.collection.get_names()
|
|
self.assertEqual(len(names), 3,
|
|
"Expected 3 objects, found %d" % len(names))
|
|
|
|
# -------------------------------------------------------
|
|
# Make sure the CNC Job Object has the correct name
|
|
# -------------------------------------------------------
|
|
cnc_name = geo_name + "_cnc"
|
|
self.assertTrue(cnc_name in names,
|
|
"Object named %s not found." % geo_name)
|
|
|
|
# -------------------------------------------------------
|
|
# Get the object make sure it's a CNC Job object
|
|
# -------------------------------------------------------
|
|
cnc_obj = self.fc.collection.get_by_name(cnc_name)
|
|
self.assertTrue(isinstance(cnc_obj, FlatCAMCNCjob),
|
|
"Expected a FlatCAMCNCJob, got %s" % type(geo_obj))
|
|
|
|
# -----------------------------------------
|
|
# Export G-Code, check output
|
|
# -----------------------------------------
|
|
assert isinstance(cnc_obj, FlatCAMCNCjob)
|
|
output_filename = ""
|
|
# get system temporary file(try create it and delete also)
|
|
with tempfile.NamedTemporaryFile(prefix='unittest.',
|
|
suffix="." + cnc_name + '.gcode',
|
|
delete=True) as tmp_file:
|
|
output_filename = tmp_file.name
|
|
cnc_obj.export_gcode(output_filename)
|
|
self.assertTrue(os.path.isfile(output_filename))
|
|
os.remove(output_filename)
|
|
|
|
print(names)
|