72 lines
2.7 KiB
Python
72 lines
2.7 KiB
Python
import os, shutil
|
|
import unittest
|
|
|
|
unittest.TestLoader.sortTestMethodsUsing = None
|
|
|
|
class TestShaping(unittest.TestCase):
|
|
def setUp(self):
|
|
try:
|
|
import netgen.occ
|
|
_ = netgen.occ.Pnt(0, 0, 0)
|
|
|
|
except Exception as e:
|
|
self.skipTest(e)
|
|
|
|
else:
|
|
from anisotropy import shaping
|
|
|
|
self.shaping = shaping
|
|
|
|
self.outputPath = os.path.join(os.path.dirname(__file__), "test_shaping_output")
|
|
os.makedirs(self.outputPath, exist_ok = True)
|
|
|
|
def test_simple(self):
|
|
simple100 = self.shaping.Simple(direction = [1, 0, 0], alpha = 0.01)
|
|
simple001 = self.shaping.Simple(direction = [0, 0, 1], alpha = 0.01)
|
|
simple111 = self.shaping.Simple(direction = [1, 1, 1], alpha = 0.01)
|
|
|
|
simple100.build()
|
|
simple001.build()
|
|
simple111.build()
|
|
|
|
simple100.export(os.path.join(self.outputPath, "simple100.step"))
|
|
simple001.export(os.path.join(self.outputPath, "simple001.step"))
|
|
simple111.export(os.path.join(self.outputPath, "simple111.step"))
|
|
|
|
def test_bodyCentered(self):
|
|
bodyCentered100 = self.shaping.BodyCentered(direction = [1, 0, 0], alpha = 0.01)
|
|
bodyCentered001 = self.shaping.BodyCentered(direction = [0, 0, 1], alpha = 0.01)
|
|
bodyCentered111 = self.shaping.BodyCentered(direction = [1, 1, 1], alpha = 0.01)
|
|
|
|
bodyCentered100.build()
|
|
bodyCentered001.build()
|
|
bodyCentered111.build()
|
|
|
|
bodyCentered100.export(os.path.join(self.outputPath, "bodyCentered100.step"))
|
|
bodyCentered001.export(os.path.join(self.outputPath, "bodyCentered001.step"))
|
|
bodyCentered111.export(os.path.join(self.outputPath, "bodyCentered111.step"))
|
|
|
|
def test_faceCentered_lattice(self):
|
|
fc = self.shaping.FaceCentered([1, 0, 0], alpha = 0.01, filletsEnabled = True)
|
|
fc.build()
|
|
fc.lattice.WriteStep(os.path.join(self.outputPath, "fc_lattice.step"))
|
|
|
|
def test_faceCentered(self):
|
|
faceCentered100 = self.shaping.FaceCentered(direction = [1, 0, 0], alpha = 0.01)
|
|
faceCentered001 = self.shaping.FaceCentered(direction = [0, 0, 1], alpha = 0.01)
|
|
faceCentered111 = self.shaping.FaceCentered(direction = [1, 1, 1], alpha = 0.01)
|
|
|
|
faceCentered100.build()
|
|
faceCentered001.build()
|
|
faceCentered111.build()
|
|
|
|
faceCentered100.export(os.path.join(self.outputPath, "faceCentered100.step"))
|
|
faceCentered001.export(os.path.join(self.outputPath, "faceCentered001.step"))
|
|
faceCentered111.export(os.path.join(self.outputPath, "faceCentered111.step"))
|
|
|
|
def tearDown(self):
|
|
shutil.rmtree(self.outputPath)
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|