#! /usr/bin/env python
import os
import tempfile
import unittest
from pathlib import Path

from solid.objects import background, circle, cube, cylinder, debug, disable
from solid.objects import hole, import_scad, include, part, root, rotate, sphere
from solid.objects import square, translate, use, color, polygon

# NOTE: the following impports aren't explicitly tested
# from solid.objects import difference, hull
# from solid.objects import (
#     import_,
#     intersection,
#     intersection_for,
#     linear_extrude,
#     import_dxf,
# )
# from solid.objects import import_stl, minkowski, mirror, multmatrix, offset, polygon
# from solid.objects import polyhedron, projection, render, resize, rotate_extrude
# from solid.objects import scale, surface, union

from solid.solidpython import (
    scad_render,
    scad_render_animated_file,
    scad_render_to_file,
)
from solid.test.ExpandedTestCase import DiffOutput

scad_test_case_templates = [
    {
        "name": "polygon",
        "class": "polygon",
        "kwargs": {"paths": [[0, 1, 2]]},
        "expected": "\n\npolygon(paths = [[0, 1, 2]], points = [[0, 0], [1, 0], [0, 1]]);",
        "args": {"points": [[0, 0, 0], [1, 0, 0], [0, 1, 0]]},
    },
    {
        "name": "polygon",
        "class": "polygon",
        "kwargs": {},
        "expected": "\n\npolygon(points = [[0, 0], [1, 0], [0, 1]]);",
        "args": {"points": [[0, 0, 0], [1, 0, 0], [0, 1, 0]]},
    },
    {
        "name": "polygon",
        "class": "polygon",
        "kwargs": {},
        "expected": "\n\npolygon(convexity = 3, points = [[0, 0], [1, 0], [0, 1]]);",
        "args": {"points": [[0, 0, 0], [1, 0, 0], [0, 1, 0]], "convexity": 3},
    },
    {
        "name": "circle",
        "class": "circle",
        "kwargs": {"segments": 12, "r": 1},
        "expected": "\n\ncircle($fn = 12, r = 1);",
        "args": {},
    },
    {
        "name": "circle_diam",
        "class": "circle",
        "kwargs": {"segments": 12, "d": 1},
        "expected": "\n\ncircle($fn = 12, d = 1);",
        "args": {},
    },
    {
        "name": "square",
        "class": "square",
        "kwargs": {"center": False, "size": 1},
        "expected": "\n\nsquare(center = false, size = 1);",
        "args": {},
    },
    {
        "name": "sphere",
        "class": "sphere",
        "kwargs": {"segments": 12, "r": 1},
        "expected": "\n\nsphere($fn = 12, r = 1);",
        "args": {},
    },
    {
        "name": "sphere_diam",
        "class": "sphere",
        "kwargs": {"segments": 12, "d": 1},
        "expected": "\n\nsphere($fn = 12, d = 1);",
        "args": {},
    },
    {
        "name": "cube",
        "class": "cube",
        "kwargs": {"center": False, "size": 1},
        "expected": "\n\ncube(center = false, size = 1);",
        "args": {},
    },
    {
        "name": "cylinder",
        "class": "cylinder",
        "kwargs": {
            "r1": None,
            "r2": None,
            "h": 1,
            "segments": 12,
            "r": 1,
            "center": False,
        },
        "expected": "\n\ncylinder($fn = 12, center = false, h = 1, r = 1);",
        "args": {},
    },
    {
        "name": "cylinder_d1d2",
        "class": "cylinder",
        "kwargs": {"d1": 4, "d2": 2, "h": 1, "segments": 12, "center": False},
        "expected": "\n\ncylinder($fn = 12, center = false, d1 = 4, d2 = 2, h = 1);",
        "args": {},
    },
    {
        "name": "polyhedron",
        "class": "polyhedron",
        "kwargs": {"convexity": None},
        "expected": "\n\npolyhedron(faces = [[0, 1, 2]], points = [[0, 0, 0], [1, 0, 0], [0, 1, 0]]);",
        "args": {"points": [[0, 0, 0], [1, 0, 0], [0, 1, 0]], "faces": [[0, 1, 2]]},
    },
    {
        "name": "polyhedron_default_convexity",
        "class": "polyhedron",
        "kwargs": {},
        "expected": "\n\npolyhedron(convexity = 10, faces = [[0, 1, 2]], points = [[0, 0, 0], [1, 0, 0], [0, 1, 0]]);",
        "args": {"points": [[0, 0, 0], [1, 0, 0], [0, 1, 0]], "faces": [[0, 1, 2]]},
    },
    {
        "name": "union",
        "class": "union",
        "kwargs": {},
        "expected": "\n\nunion();",
        "args": {},
    },
    {
        "name": "intersection",
        "class": "intersection",
        "kwargs": {},
        "expected": "\n\nintersection();",
        "args": {},
    },
    {
        "name": "difference",
        "class": "difference",
        "kwargs": {},
        "expected": "\n\ndifference();",
        "args": {},
    },
    {
        "name": "translate",
        "class": "translate",
        "kwargs": {"v": [1, 0, 0]},
        "expected": "\n\ntranslate(v = [1, 0, 0]);",
        "args": {},
    },
    {
        "name": "scale",
        "class": "scale",
        "kwargs": {"v": 0.5},
        "expected": "\n\nscale(v = 0.5000000000);",
        "args": {},
    },
    {
        "name": "rotate",
        "class": "rotate",
        "kwargs": {"a": 45, "v": [0, 0, 1]},
        "expected": "\n\nrotate(a = 45, v = [0, 0, 1]);",
        "args": {},
    },
    {
        "name": "mirror",
        "class": "mirror",
        "kwargs": {},
        "expected": "\n\nmirror(v = [0, 0, 1]);",
        "args": {"v": [0, 0, 1]},
    },
    {
        "name": "resize",
        "class": "resize",
        "kwargs": {"newsize": [5, 5, 5], "auto": [True, True, False]},
        "expected": "\n\nresize(auto = [true, true, false], newsize = [5, 5, 5]);",
        "args": {},
    },
    {
        "name": "multmatrix",
        "class": "multmatrix",
        "kwargs": {},
        "expected": "\n\nmultmatrix(m = [[1, 0, 0, 0], [0, 1, 0, 0], [0, 0, 1, 0], [0, 0, 0, 1]]);",
        "args": {"m": [[1, 0, 0, 0], [0, 1, 0, 0], [0, 0, 1, 0], [0, 0, 0, 1]]},
    },
    {
        "name": "minkowski",
        "class": "minkowski",
        "kwargs": {},
        "expected": "\n\nminkowski();",
        "args": {},
    },
    {
        "name": "offset",
        "class": "offset",
        "kwargs": {"r": 1},
        "expected": "\n\noffset(r = 1);",
        "args": {},
    },
    {
        "name": "offset_segments",
        "class": "offset",
        "kwargs": {"r": 1, "segments": 12},
        "expected": "\n\noffset($fn = 12, r = 1);",
        "args": {},
    },
    {
        "name": "offset_chamfer",
        "class": "offset",
        "kwargs": {"delta": 1},
        "expected": "\n\noffset(chamfer = false, delta = 1);",
        "args": {},
    },
    {
        "name": "offset_zero_delta",
        "class": "offset",
        "kwargs": {"r": 0},
        "expected": "\n\noffset(r = 0);",
        "args": {},
    },
    {
        "name": "hull",
        "class": "hull",
        "kwargs": {},
        "expected": "\n\nhull();",
        "args": {},
    },
    {
        "name": "render",
        "class": "render",
        "kwargs": {"convexity": None},
        "expected": "\n\nrender();",
        "args": {},
    },
    {
        "name": "projection",
        "class": "projection",
        "kwargs": {"cut": None},
        "expected": "\n\nprojection();",
        "args": {},
    },
    {
        "name": "surface",
        "class": "surface",
        "kwargs": {"center": False, "convexity": None},
        "expected": '\n\nsurface(center = false, file = "/Path/to/dummy.dxf");',
        "args": {"file": "'/Path/to/dummy.dxf'"},
    },
    {
        "name": "import_stl",
        "class": "import_stl",
        "kwargs": {"layer": None, "origin": (0, 0)},
        "expected": '\n\nimport(file = "/Path/to/dummy.stl", origin = [0, 0]);',
        "args": {"file": "'/Path/to/dummy.stl'"},
    },
    {
        "name": "import_dxf",
        "class": "import_dxf",
        "kwargs": {"layer": None, "origin": (0, 0)},
        "expected": '\n\nimport(file = "/Path/to/dummy.dxf", origin = [0, 0]);',
        "args": {"file": "'/Path/to/dummy.dxf'"},
    },
    {
        "name": "import_",
        "class": "import_",
        "kwargs": {"layer": None, "origin": (0, 0)},
        "expected": '\n\nimport(file = "/Path/to/dummy.dxf", origin = [0, 0]);',
        "args": {"file": "'/Path/to/dummy.dxf'"},
    },
    {
        "name": "import__convexity",
        "class": "import_",
        "kwargs": {"layer": None, "origin": (0, 0), "convexity": 2},
        "expected": '\n\nimport(convexity = 2, file = "/Path/to/dummy.dxf", origin = [0, 0]);',
        "args": {"file": "'/Path/to/dummy.dxf'"},
    },
    {
        "name": "linear_extrude",
        "class": "linear_extrude",
        "kwargs": {
            "twist": None,
            "slices": None,
            "center": False,
            "convexity": None,
            "height": 1,
            "scale": 0.9,
        },
        "expected": "\n\nlinear_extrude(center = false, height = 1, scale = 0.9000000000);",
        "args": {},
    },
    {
        "name": "rotate_extrude",
        "class": "rotate_extrude",
        "kwargs": {"angle": 90, "segments": 4, "convexity": None},
        "expected": "\n\nrotate_extrude($fn = 4, angle = 90);",
        "args": {},
    },
    {
        "name": "intersection_for",
        "class": "intersection_for",
        "kwargs": {},
        "expected": "\n\nintersection_for(n = [0, 1, 2]);",
        "args": {"n": [0, 1, 2]},
    },
]


class TemporaryFileBuffer(object):
    name = None
    contents = None

    def __enter__(self):
        f = tempfile.NamedTemporaryFile(delete=False)
        self.name = f.name
        try:
            f.close()
        except:
            self._cleanup()
            raise
        return self

    def __exit__(self, exc_type, exc_val, exc_tb):
        try:
            with open(self.name, "r") as f:
                self.contents = f.read()
        finally:
            self._cleanup()

    def _cleanup(self):
        try:
            os.unlink(self.name)
        except Exception:
            pass


class TestSolidPython(DiffOutput):
    # test cases will be dynamically added to this instance

    def expand_scad_path(self, filename):
        path = Path(__file__).absolute().parent.parent / filename
        return path

    def test_infix_union(self):
        a = cube(2)
        b = sphere(2)
        expected = "\n\nunion() {\n\tcube(size = 2);\n\tsphere(r = 2);\n}"
        actual = scad_render(a + b)
        self.assertEqual(expected, actual)

    def test_infix_difference(self):
        a = cube(2)
        b = sphere(2)
        expected = "\n\ndifference() {\n\tcube(size = 2);\n\tsphere(r = 2);\n}"
        actual = scad_render(a - b)
        self.assertEqual(expected, actual)

    def test_infix_intersection(self):
        a = cube(2)
        b = sphere(2)
        expected = "\n\nintersection() {\n\tcube(size = 2);\n\tsphere(r = 2);\n}"
        actual = scad_render(a * b)
        self.assertEqual(expected, actual)

    def test_parse_scad_callables(self):
        test_str = """
                    module hex (width=10, height=10,
                                flats= true, center=false){}
                    function righty (angle=90) = 1;
                    function lefty(avar) = 2;
                    module more(a=[something, other]) {}
                    module pyramid(side=10, height=-1, square=false, centerHorizontal=true, centerVertical=false){}
                    module no_comments(arg=10,   //test comment
                    other_arg=2, /* some extra comments
                    on empty lines */
                    last_arg=4){}
                    module float_arg(arg=1.0){}
                    module arg_var(var5){}
                    module kwarg_var(var2=78){}
                    module var_true(var_true = true){}
                    module var_false(var_false = false){}
                    module var_int(var_int = 5){}
                    module var_negative(var_negative = -5){}
                    module var_float(var_float = 5.5){}
                    module var_number(var_number = -5e89){}
                    module var_empty_vector(var_empty_vector = []){}
                    module var_simple_string(var_simple_string = "simple string"){}
                    module var_complex_string(var_complex_string = "a \\"complex\\"\\tstring with a\\\\"){}
                    module var_vector(var_vector = [5454445, 565, [44545]]){}
                    module var_complex_vector(var_complex_vector = [545 + 4445, 565, [cos(75) + len("yes", 45)]]){}
                    module var_vector(var_vector = [5, 6, "string\\twith\\ttab"]){}
                    module var_range(var_range = [0:10e10]){}
                    module var_range_step(var_range_step = [-10:0.5:10]){}
                    module var_with_arithmetic(var_with_arithmetic = 8 * 9 - 1 + 89 / 15){}
                    module var_with_parentheses(var_with_parentheses = 8 * ((9 - 1) + 89) / 15){}
                    module var_with_functions(var_with_functions = abs(min(chamferHeight2, 0)) / 1){}
                    module var_with_conditional_assignment(var_with_conditional_assignment = mytest ? 45 : yop){}

                   """

        scad_file = ""
        with tempfile.NamedTemporaryFile(suffix=".scad", delete=False) as f:
            f.write(test_str.encode("utf-8"))
            scad_file = f.name

        expected = [
            {
                "name": "hex",
                "args": [],
                "kwargs": ["width", "height", "flats", "center"],
            },
            {"name": "righty", "args": [], "kwargs": ["angle"]},
            {"name": "lefty", "args": [], "kwargs": ["avar"]},
            {"name": "more", "args": [], "kwargs": ["a"]},
            {
                "name": "pyramid",
                "args": [],
                "kwargs": [
                    "side",
                    "height",
                    "square",
                    "centerHorizontal",
                    "centerVertical",
                ],
            },
            {
                "name": "no_comments",
                "args": [],
                "kwargs": ["arg", "other_arg", "last_arg"],
            },
            {"name": "float_arg", "args": [], "kwargs": ["arg"]},
            {"name": "arg_var", "args": [], "kwargs": ["var5"]},
            {"name": "kwarg_var", "args": [], "kwargs": ["var2"]},
            {"name": "var_true", "args": [], "kwargs": ["var_true"]},
            {"name": "var_false", "args": [], "kwargs": ["var_false"]},
            {"name": "var_int", "args": [], "kwargs": ["var_int"]},
            {"name": "var_negative", "args": [], "kwargs": ["var_negative"]},
            {"name": "var_float", "args": [], "kwargs": ["var_float"]},
            {"name": "var_number", "args": [], "kwargs": ["var_number"]},
            {"name": "var_empty_vector", "args": [], "kwargs": ["var_empty_vector"]},
            {"name": "var_simple_string", "args": [], "kwargs": ["var_simple_string"]},
            {
                "name": "var_complex_string",
                "args": [],
                "kwargs": ["var_complex_string"],
            },
            {"name": "var_vector", "args": [], "kwargs": ["var_vector"]},
            {
                "name": "var_complex_vector",
                "args": [],
                "kwargs": ["var_complex_vector"],
            },
            {"name": "var_vector", "args": [], "kwargs": ["var_vector"]},
            {"name": "var_range", "args": [], "kwargs": ["var_range"]},
            {"name": "var_range_step", "args": [], "kwargs": ["var_range_step"]},
            {
                "name": "var_with_arithmetic",
                "args": [],
                "kwargs": ["var_with_arithmetic"],
            },
            {
                "name": "var_with_parentheses",
                "args": [],
                "kwargs": ["var_with_parentheses"],
            },
            {
                "name": "var_with_functions",
                "args": [],
                "kwargs": ["var_with_functions"],
            },
            {
                "name": "var_with_conditional_assignment",
                "args": [],
                "kwargs": ["var_with_conditional_assignment"],
            },
        ]

        from solid.solidpython import parse_scad_callables

        actual = parse_scad_callables(scad_file)

        for e in expected:
            self.assertEqual(e in actual, True)

        os.unlink(scad_file)

    def test_use(self):
        include_file = self.expand_scad_path("examples/scad_to_include.scad")
        use(include_file)

        a = steps(3)  # type: ignore # noqa: F821
        actual = scad_render(a)

        abs_path = a._get_include_path(include_file)
        expected = f"use <{abs_path}>\n\n\nsteps(howmany = 3);"
        self.assertEqual(expected, actual)

    def test_import_scad(self):
        include_file = self.expand_scad_path("examples/scad_to_include.scad")
        mod = import_scad(include_file)
        a = mod.steps(3)
        actual = scad_render(a)

        abs_path = a._get_include_path(include_file)
        expected = f"use <{abs_path}>\n\n\nsteps(howmany = 3);"
        self.assertEqual(expected, actual)

        # Make sure this plays nicely with `scad_render()`'s `file_header` arg
        header = "$fn = 24;"
        actual = scad_render(a, file_header=header)
        expected = f"{header}\nuse <{abs_path}>\n\n\nsteps(howmany = 3);"
        self.assertEqual(expected, actual)

        # Confirm that we can leave out even non-default arguments in OpenSCAD
        a = mod.optional_nondefault_arg()
        actual = scad_render(a)
        expected = f"use <{abs_path}>\n\n\noptional_nondefault_arg();"
        self.assertEqual(expected, actual)
        # Make sure we throw ValueError on nonexistent imports
        self.assertRaises(ValueError, import_scad, "path/doesnt/exist.scad")

        # Test that we recursively import directories correctly
        examples = import_scad(include_file.parent)
        self.assertTrue(hasattr(examples, "scad_to_include"))
        self.assertTrue(hasattr(examples.scad_to_include, "steps"))

        # Test that:
        # A) scad files in the designated OpenSCAD library directories
        #       (path-dependent, see: solid.objects._openscad_library_paths())
        #       are imported correctly.
        # B) scad files in the designated app-install library directories
        from solid import objects

        lib_dirs = objects._openscad_library_paths()
        for i, ld in enumerate(lib_dirs):
            if ld.as_posix() == ".":
                continue
            if not ld.exists():
                continue
            temp_dirname = f"test_{i}"
            d = ld / temp_dirname
            try:
                d.mkdir(exist_ok=True)
            except PermissionError:
                # We won't always have permissions to write to the library directory.
                # In that case, skip this test.
                continue
            p = d / "scad_to_include.scad"
            p.write_text(include_file.read_text())
            temp_file_str = f"{temp_dirname}/scad_to_include.scad"

            mod = import_scad(temp_file_str)
            a = mod.steps(3)
            actual = scad_render(a)
            expected = f"use <{p.absolute()}>\n\n\nsteps(howmany = 3);"
            self.assertEqual(
                actual, expected, f"Unexpected file contents at {p} for dir: {ld}"
            )

            # remove generated file and directories
            p.unlink()
            d.rmdir()

    def test_multiple_import_scad(self):
        # For Issue #172. Originally, multiple `import_scad()` calls would
        # re-import the entire module, rather than cache a module after one use
        include_file = self.expand_scad_path("examples/scad_to_include.scad")
        mod1 = import_scad(include_file)
        mod2 = import_scad(include_file)
        self.assertEqual(mod1, mod2)

    def test_imported_scad_arguments(self):
        include_file = self.expand_scad_path("examples/scad_to_include.scad")
        mod = import_scad(include_file)
        points = mod.scad_points()
        poly = polygon(points)
        actual = scad_render(poly)
        abs_path = points._get_include_path(include_file)
        expected = f"use <{abs_path}>\n\n\npolygon(points = scad_points());"
        self.assertEqual(expected, actual)

    def test_use_reserved_words(self):
        scad_str = """module reserved_word_arg(or=3){\n\tcube(or);\n}\nmodule or(arg=3){\n\tcube(arg);\n}\n"""

        fd, path = tempfile.mkstemp(text=True)
        try:
            os.close(fd)
            with open(path, "w") as f:
                f.write(scad_str)

            use(path)
            a = reserved_word_arg(or_=5)  # type: ignore # noqa: F821
            actual = scad_render(a)
            expected = f"use <{path}>\n\n\nreserved_word_arg(or = 5);"
            self.assertEqual(expected, actual)

            b = or_(arg=5)  # type: ignore # noqa: F821
            actual = scad_render(b)
            expected = f"use <{path}>\n\n\nor(arg = 5);"
            self.assertEqual(expected, actual)
        finally:
            os.remove(path)

    def test_include(self):
        include_file = self.expand_scad_path("examples/scad_to_include.scad")
        self.assertIsNotNone(include_file, "examples/scad_to_include.scad not found")
        include(include_file)
        a = steps(3)  # type: ignore # noqa: F821

        actual = scad_render(a)
        abs_path = a._get_include_path(include_file)
        expected = f"include <{abs_path}>\n\n\nsteps(howmany = 3);"
        self.assertEqual(expected, actual)

    def test_extra_args_to_included_scad(self):
        include_file = self.expand_scad_path("examples/scad_to_include.scad")
        mod = import_scad(include_file)
        a = mod.steps(3, external_var=True)
        actual = scad_render(a)

        abs_path = a._get_include_path(include_file)
        expected = f"use <{abs_path}>\n\n\nsteps(external_var = true, howmany = 3);"
        self.assertEqual(expected, actual)

    def test_background(self):
        a = cube(10)
        expected = "\n\n%cube(size = 10);"
        actual = scad_render(background(a))
        self.assertEqual(expected, actual)

    def test_debug(self):
        a = cube(10)
        expected = "\n\n#cube(size = 10);"
        actual = scad_render(debug(a))
        self.assertEqual(expected, actual)

    def test_disable(self):
        a = cube(10)
        expected = "\n\n*cube(size = 10);"
        actual = scad_render(disable(a))
        self.assertEqual(expected, actual)

    def test_root(self):
        a = cube(10)
        expected = "\n\n!cube(size = 10);"
        actual = scad_render(root(a))
        self.assertEqual(expected, actual)

    def test_color(self):
        all_args = [
            {"c": [1, 0, 0]},
            {"c": [1, 0, 0], "alpha": 0.5},
            {"c": "#66F"},
            {"c": "Teal", "alpha": 0.5},
        ]

        expecteds = [
            "\n\ncolor(alpha = 1.0000000000, c = [1, 0, 0]);",
            "\n\ncolor(alpha = 0.5000000000, c = [1, 0, 0]);",
            '\n\ncolor(alpha = 1.0000000000, c = "#66F");',
            '\n\ncolor(alpha = 0.5000000000, c = "Teal");',
        ]
        for args, expected in zip(all_args, expecteds):
            col = color(**args)
            actual = scad_render(col)
            self.assertEqual(expected, actual)

    def test_explicit_hole(self):
        a = cube(10, center=True) + hole()(cylinder(2, 20, center=True))
        expected = "\n\ndifference(){\n\tunion() {\n\t\tcube(center = true, size = 10);\n\t}\n\t/* Holes Below*/\n\tunion(){\n\t\tcylinder(center = true, h = 20, r = 2);\n\t} /* End Holes */ \n}"
        actual = scad_render(a)
        self.assertEqual(expected, actual)

    def test_hole_transform_propagation(self):
        # earlier versions of holes had problems where a hole
        # that was used a couple places wouldn't propagate correctly.
        # Confirm that's still happening as it's supposed to
        h = hole()(rotate(a=90, v=[0, 1, 0])(cylinder(2, 20, center=True)))

        h_vert = rotate(a=-90, v=[0, 1, 0])(h)

        a = cube(10, center=True) + h + h_vert
        expected = "\n\ndifference(){\n\tunion() {\n\t\tcube(center = true, size = 10);\n\t\trotate(a = -90, v = [0, 1, 0]) {\n\t\t}\n\t}\n\t/* Holes Below*/\n\tunion(){\n\t\trotate(a = 90, v = [0, 1, 0]) {\n\t\t\tcylinder(center = true, h = 20, r = 2);\n\t\t}\n\t\trotate(a = -90, v = [0, 1, 0]){\n\t\t\trotate(a = 90, v = [0, 1, 0]) {\n\t\t\t\tcylinder(center = true, h = 20, r = 2);\n\t\t\t}\n\t\t}\n\t} /* End Holes */ \n}"
        actual = scad_render(a)
        self.assertEqual(expected, actual)

    def test_separate_part_hole(self):
        # Make two parts, a block with hole, and a cylinder that
        # fits inside it.  Make them separate parts, meaning
        # holes will be defined at the level of the part_root node,
        # not the overall node.  This allows us to preserve holes as
        # first class space, but then to actually fill them in with
        # the parts intended to fit in them.
        b = cube(10, center=True)
        c = cylinder(r=2, h=12, center=True)
        p1 = b - hole()(c)

        # Mark this cube-with-hole as a separate part from the cylinder
        p1 = part()(p1)

        # This fits in the hole.  If p1 is set as a part_root, it will all appear.
        # If not, the portion of the cylinder inside the cube will not appear,
        # since it would have been removed by the hole in p1
        p2 = cylinder(r=1.5, h=14, center=True)

        a = p1 + p2

        expected = "\n\nunion() {\n\tdifference(){\n\t\tdifference() {\n\t\t\tcube(center = true, size = 10);\n\t\t}\n\t\t/* Holes Below*/\n\t\tunion(){\n\t\t\tcylinder(center = true, h = 12, r = 2);\n\t\t} /* End Holes */ \n\t}\n\tcylinder(center = true, h = 14, r = 1.5000000000);\n}"
        actual = scad_render(a)
        self.assertEqual(expected, actual)

    def test_scad_render_animated_file(self):
        def my_animate(_time=0):
            import math

            # _time will range from 0 to 1, not including 1
            rads = _time * 2 * math.pi
            rad = 15
            c = translate([rad * math.cos(rads), rad * math.sin(rads)])(square(10))
            return c

        with TemporaryFileBuffer() as tmp:
            scad_render_animated_file(
                my_animate,
                steps=2,
                back_and_forth=False,
                filepath=tmp.name,
                include_orig_code=False,
            )

        actual = tmp.contents
        expected = "\nif ($t >= 0.0 && $t < 0.5){   \n\ttranslate(v = [15.0000000000, 0.0000000000]) {\n\t\tsquare(size = 10);\n\t}\n}\nif ($t >= 0.5 && $t < 1.0){   \n\ttranslate(v = [-15.0000000000, 0.0000000000]) {\n\t\tsquare(size = 10);\n\t}\n}\n"

        self.assertEqual(expected, actual)

    def test_scad_render_to_file(self):
        a = circle(10)

        # No header, no included original code
        with TemporaryFileBuffer() as tmp:
            scad_render_to_file(a, filepath=tmp.name, include_orig_code=False)

        actual = tmp.contents
        expected = "\n\ncircle(r = 10);"

        # scad_render_to_file also adds a date & version stamp before scad code;
        # That won't match here, so just make sure the correct code is at the end
        self.assertTrue(actual.endswith(expected))

        # Header
        with TemporaryFileBuffer() as tmp:
            scad_render_to_file(
                a, filepath=tmp.name, include_orig_code=False, file_header="$fn = 24;"
            )

        actual = tmp.contents
        expected = "$fn = 24;\n\n\ncircle(r = 10);"

        self.assertTrue(actual.endswith(expected))

        # Test out_dir specification, both using an existing dir & creating one
        # Using existing directory
        with TemporaryFileBuffer() as tmp:
            out_dir = Path(tmp.name).parent
            expected = (out_dir / "test_solidpython.scad").as_posix()
            actual = scad_render_to_file(a, out_dir=out_dir)
            self.assertEqual(expected, actual)

        # Creating a directory on demand
        with TemporaryFileBuffer() as tmp:
            out_dir = Path(tmp.name).parent / "SCAD"
            expected = (out_dir / "test_solidpython.scad").as_posix()
            actual = scad_render_to_file(a, out_dir=out_dir)
            self.assertEqual(expected, actual)

        # TODO: test include_orig_code=True, but that would have to
        # be done from a separate file, or include everything in this one

    def test_numpy_type(self):
        try:
            import numpy  # type: ignore

            numpy_cube = cube(size=numpy.array([1, 2, 3]))
            expected = "\n\ncube(size = [1,2,3]);"
            actual = scad_render(numpy_cube)
            self.assertEqual(
                expected, actual, "Numpy SolidPython not rendered correctly"
            )
        except ImportError:
            pass

    def test_custom_iterables(self):
        from euclid3 import Vector3

        class CustomIterable:
            def __iter__(self):
                return iter([1, 2, 3])

        expected = "\n\ncube(size = [1, 2, 3]);"
        iterables = [
            [1, 2, 3],
            (1, 2, 3),
            Vector3(1, 2, 3),
            CustomIterable(),
        ]

        for iterable in iterables:
            name = type(iterable).__name__
            actual = scad_render(cube(size=iterable))
            self.assertEqual(
                expected, actual, f"{name} SolidPython not rendered correctly"
            )


def single_test(test_dict):
    _, cls, args, kwargs, expected = (
        test_dict["name"],
        test_dict["class"],
        test_dict["args"],
        test_dict["kwargs"],
        test_dict["expected"],
    )

    def test(self):
        call_str = cls + "("
        for k, v in args.items():
            call_str += f"{k}={v}, "
        for k, v in kwargs.items():
            call_str += f"{k}={v}, "
        call_str += ")"

        scad_obj = eval(call_str)
        actual = scad_render(scad_obj)

        self.assertEqual(expected, actual)

    return test


def generate_cases_from_templates():
    for test_dict in scad_test_case_templates:
        test = single_test(test_dict)
        test_name = f"test_{test_dict['name']}"
        setattr(TestSolidPython, test_name, test)


if __name__ == "__main__":
    generate_cases_from_templates()
    unittest.main()