gridfinity-rebuilt-openscad/tests/test_spiral_vase.py

"""
Tests for gridfinity-spiral-vase.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 TestSpiralVaseBase(unittest.TestCase):
    """
    Test creating a single base in "gridfinity-spiral-vase.scad"

    Currently only makes sure code runs, and outputs pictures for manual verification.
    """

    @classmethod
    def setUpClass(cls):
        parameter_file_path = Path("gridfinity-spiral-vase.json")
        parameter_file_data = ParameterFile.from_json(parameter_file_path.read_text())
        cls.default_parameters = parameter_file_data.parameterSets["Default"]

    def setUp(self):
        self.scad_runner = OpenScadRunner(Path('../gridfinity-spiral-vase.scad'))
        self.scad_runner.image_folder_base = Path('../images/spiral_vase_base/')
        self.scad_runner.parameters = self.default_parameters.copy()
        self.scad_runner.parameters["type"] = 1 # Create a Base
        self.scad_runner.camera_arguments = CameraArguments(Vec3(0,0,0), CameraRotations.AngledBottom, 150)

    def test_no_holes(self):
        vars = self.scad_runner.parameters
        vars["enable_holes"] = False
        self.scad_runner.create_image([], Path('no_holes_bottom.png'))
        self.scad_runner.camera_arguments = self.scad_runner.camera_arguments.with_rotation(CameraRotations.Top)
        self.scad_runner.create_image([], Path('no_holes_top.png'))

    def test_refined_holes(self):
        vars = self.scad_runner.parameters
        vars["enable_holes"] = True
        self.scad_runner.create_image([], Path('with_holes_bottom.png'))
        self.scad_runner.camera_arguments = self.scad_runner.camera_arguments.with_rotation(CameraRotations.Top)
        self.scad_runner.create_image([], Path('with_holes_top.png'))


if __name__ == '__main__':
    unittest.main()