gridfinity-rebuilt-openscad/tests/test_bins.py
Arthur Moore 9b89773812 Add a test suite to generate images of bins with holes
Used to ensure all hole options work.
2024-04-28 02:05:50 -04:00

142 lines
5.4 KiB
Python

"""
Tests for gridfinity-rebuilt-bins.scad
@Copyright Arthur Moore 2024 MIT License
"""
import dataclasses
import json
import unittest
from pathlib import Path
from tempfile import NamedTemporaryFile
from openscad_runner import *
class TestBinHoles(unittest.TestCase):
"""
Test how a single base looks with holes cut out.
Currently only makes sure code runs, and outputs pictures for manual verification.
"""
@classmethod
def setUpClass(cls):
parameter_file_path = Path("gridfinity-rebuilt-bins.json")
parameter_file_data = ParameterFile.from_json(parameter_file_path.read_text())
cls.default_parameters = parameter_file_data.parameterSets["Default"]
cls.camera_args = CameraArguments(Vec3(0,0,0),Vec3(225,0,225),150)
def setUp(self):
self.scad_runner = OpenScadRunner(Path('../gridfinity-rebuilt-bins.scad'))
self.scad_runner.image_folder_base = Path('../images/base_hole_options/')
self.scad_runner.parameters = self.default_parameters.copy()
def test_no_holes(self):
vars = self.scad_runner.parameters
vars["refined_hole"] = False
vars["magnet_hole"] = False
vars["screw_hole"] = False
self.scad_runner.create_image(self.camera_args, [], Path('no_holes.png'))
def test_refined_holes(self):
vars = self.scad_runner.parameters
vars["refined_hole"] = True
vars["magnet_hole"] = False
vars["screw_hole"] = False
self.scad_runner.create_image(self.camera_args, [], Path('refined_holes.png'))
def test_refined_and_screw_holes(self):
vars = self.scad_runner.parameters
vars["refined_hole"] = True
vars["magnet_hole"] = False
vars["screw_hole"] = True
vars["printable_hole_top"] = False
self.scad_runner.create_image(self.camera_args, [], Path('refined_and_screw_holes.png'))
def test_screw_holes_plain(self):
vars = self.scad_runner.parameters
vars["refined_hole"] = False
vars["magnet_hole"] = False
vars["screw_hole"] = True
vars["printable_hole_top"] = False
self.scad_runner.create_image(self.camera_args, [], Path('screw_holes_plain.png'))
def test_screw_holes_printable(self):
vars = self.scad_runner.parameters
vars["refined_hole"] = False
vars["magnet_hole"] = False
vars["screw_hole"] = True
vars["printable_hole_top"] = True
self.scad_runner.create_image(self.camera_args, [], Path('screw_holes_printable.png'))
def test_magnet_holes_plain(self):
vars = self.scad_runner.parameters
vars["refined_hole"] = False
vars["magnet_hole"] = True
vars["screw_hole"] = False
vars["crush_ribs"] = False
vars["chamfer_magnet_holes"] = False
vars["printable_hole_top"] = False
self.scad_runner.create_image(self.camera_args, [], Path('magnet_holes_plain.png'))
def test_magnet_holes_chamfered(self):
vars = self.scad_runner.parameters
vars["refined_hole"] = False
vars["magnet_hole"] = True
vars["screw_hole"] = False
vars["crush_ribs"] = False
vars["chamfer_magnet_holes"] = True
vars["printable_hole_top"] = False
self.scad_runner.create_image(self.camera_args, [], Path('magnet_holes_chamfered.png'))
def test_magnet_holes_printable(self):
vars = self.scad_runner.parameters
vars["refined_hole"] = False
vars["magnet_hole"] = True
vars["screw_hole"] = False
vars["crush_ribs"] = False
vars["chamfer_magnet_holes"] = False
vars["printable_hole_top"] = True
self.scad_runner.create_image(self.camera_args, [], Path('magnet_holes_printable.png'))
def test_magnet_holes_with_crush_ribs(self):
vars = self.scad_runner.parameters
vars["refined_hole"] = False
vars["magnet_hole"] = True
vars["screw_hole"] = False
vars["crush_ribs"] = True
vars["chamfer_magnet_holes"] = False
vars["printable_hole_top"] = False
self.scad_runner.create_image(self.camera_args, [], Path('magnet_holes_with_crush_ribs.png'))
def test_magnet_and_screw_holes_plain(self):
vars = self.scad_runner.parameters
vars["refined_hole"] = False
vars["magnet_hole"] = True
vars["screw_hole"] = True
vars["crush_ribs"] = False
vars["chamfer_magnet_holes"] = False
vars["printable_hole_top"] = False
self.scad_runner.create_image(self.camera_args, [], Path('magnet_and_screw_holes_plain.png'))
def test_magnet_and_screw_holes_printable(self):
vars = self.scad_runner.parameters
vars["refined_hole"] = False
vars["magnet_hole"] = True
vars["screw_hole"] = True
vars["crush_ribs"] = False
vars["chamfer_magnet_holes"] = False
vars["printable_hole_top"] = True
self.scad_runner.create_image(self.camera_args, [], Path('magnet_and_screw_holes_printable.png'))
def test_magnet_and_screw_holes_all(self):
vars = self.scad_runner.parameters
vars["refined_hole"] = False
vars["magnet_hole"] = True
vars["screw_hole"] = True
vars["crush_ribs"] = True
vars["chamfer_magnet_holes"] = True
vars["printable_hole_top"] = True
self.scad_runner.create_image(self.camera_args, [], Path('magnet_and_screw_holes_all.png'))
if __name__ == '__main__':
unittest.main()