Coverage for tests/integration/test_modelloader.py: 95%

1import unittest 

2from unittest.mock import Mock 

3from tatlin.conf.config import Config 

4from tatlin.lib.model import ModelLoader 

5 

6 

7class ModelLoaderTest(unittest.TestCase): 

8 def setUp(self): 

9 self.config = Config(".asdf") 

10 

11 def test_gcode(self): 

12 model_loader = ModelLoader("tests/fixtures/gcode/top.gcode") 

13 model, Panel = model_loader.load(self.config, Mock(), Mock()) 

14 

15 self.assertAlmostEqual(model.width, 71.73, places=2) 

16 self.assertAlmostEqual(model.height, 14.4, places=2) 

17 self.assertAlmostEqual(model.depth, 117, places=2) 

18 

19 def test_stl(self): 

20 model_loader = ModelLoader("tests/fixtures/stl/top.stl") 

21 model_loader.load(self.config, Mock(), Mock()) 

22 

23 def test_binary_stl(self): 

24 model_loader = ModelLoader("tests/fixtures/stl/cube-bin.stl") 

25 model_loader.load(self.config, Mock(), Mock()) 

26 

27 

28if __name__ == "__main__": 

29 unittest.main()