def test_processing(): image = cv2.imread("../images/buffer/output.png", cv2.IMREAD_GRAYSCALE) assert not np.array_equal( Processing.process_image('../images/buffer/new.png'), image )
def process(self): logs = Logger('log_' + datetime.strftime(datetime.now(), "%d_%m_%H_%M") + '.txt') figure_numbers = { "Bar": int(self.bar_count.get()), "Circle": int(self.circle_count.get()), "Ellipse": int(self.ellipse_count.get()), "Dots": int(self.dots_count.get()), "Popcorn": int(self.popcorn_count.get()) } writer = Writer(self.file.get()) Cleaner().clear_output_images() iterations = 1 for i in range(0, figure_numbers.get("Bar")): print("Bar: step " + str(iterations) + " of " + str(figure_numbers.get("Bar"))) bar = BarBuilder().generate_random() Plotter.plot(bar, filename='images/buffer/output.png', height=int(self.res_h.get()), width=int(self.res_w.get())) image = Processing.process_image('images/Bars/sample' + str(iterations) + '.png') writer.write_row_to_file( Converter.convert_image_to_csv_row(image, "Bar") ) iterations += 1 print("Bars created") iterations = 1 for i in range(0, figure_numbers.get("Circle")): print("Circle: step " + str(iterations) + " of " + str(figure_numbers.get("Circle"))) circle = CircleBuilder().generate_random() Plotter.plot(circle, filename='images/buffer/output.png', height=int(self.res_h.get()), width=int(self.res_w.get())) image = Processing.process_image('images/Circles/sample' + str(iterations) + '.png') writer.write_row_to_file( Converter.convert_image_to_csv_row(image, "Circle") ) iterations += 1 print("Circles created") iterations = 1 for i in range(0, figure_numbers.get("Dots")): print("Dots: step " + str(iterations) + " of " + str(figure_numbers.get("Dots"))) dots = DotsBuilder().generate_random() Plotter.plot(figures=dots, filename='images/buffer/output.png', height=int(self.res_h.get()), width=int(self.res_w.get())) image = Processing.process_image('images/Dots/sample' + str(iterations) + '.png') writer.write_row_to_file( Converter.convert_image_to_csv_row(image, "Dots") ) iterations += 1 print("Dots created") iterations = 1 for i in range(0, figure_numbers.get("Ellipse")): print("Ellipse: step " + str(iterations) + " of " + str(figure_numbers.get("Ellipse"))) ellipse = EllipseBuilder().generate_random() Plotter.plot(ellipse, filename='images/buffer/output.png', height=int(self.res_h.get()), width=int(self.res_w.get())) image = Processing.process_image('images/Ellipses/sample' + str(iterations) + '.png') writer.write_row_to_file( Converter.convert_image_to_csv_row(image, "Ellipse") ) iterations += 1 print("Ellipse created") iterations = 1 for i in range(0, figure_numbers.get("Popcorn")): print("Popcorn: step " + str(iterations) + " of " + str(figure_numbers.get("Popcorn"))) popcorn = PopcornBuilder().generate_random() Plotter.plot(figures=popcorn, filename='images/buffer/output.png', height=int(self.res_h.get()), width=int(self.res_w.get())) image = Processing.process_image('images/Popcorn/sample' + str(iterations) + '.png') writer.write_row_to_file( Converter.convert_image_to_csv_row(image, "Popcorn") ) iterations += 1 print("Popcorn created") return