Skip to content
Snippets Groups Projects
Select Git revision
  • 7c3a0b48fd2c57ed2ac32b138acd3dab7e993798
  • master default
  • Omar
  • Jack
4 results

sample_test.py

Blame
  • sample_test.py 1.68 KiB
    import unittest
    import os
    import sys
    
    class TestBasicFunction(unittest.TestCase):
    
        def SetUp(self):
            modPath = os.path.abspath(os.getcwd())
            sys.path.insert(0, modPath)
        
        def test_running(self):
            print("Running sample_test.py")
            self.assertTrue(True)
        
        def test_python_imports(self):
            import numpy, scipy, matplotlib, vtk, AmpScan.core
            s = str(type(numpy))
            self.assertEqual(s, "<class 'module'>")
            s = str(type(scipy))
            self.assertEqual(s, "<class 'module'>")
            s = str(type(matplotlib))
            self.assertEqual(s, "<class 'module'>")
            s = str(type(vtk))
            self.assertEqual(s, "<class 'module'>")
            s = str(type(AmpScan.core))
            self.assertEqual(s, "<class 'module'>", "Failed import: AmpScan.core")
    
        @unittest.expectedFailure
        def test_failure(self):
            s = str(type("string"))
            self.assertEqual(s, "<class 'module'>")
    
        def test_rotate(self):
            stlPath = os.path.abspath(os.getcwd()) + "\\tests\\sample_stl_sphere_BIN.stl"
            from AmpScan.core import AmpObject
            Amp = AmpObject(stlPath)
            s = str(type(Amp))
            self.assertEqual(s, "<class 'AmpScan.core.AmpObject'>", "Not expected Object")
            with self.assertRaises(TypeError):
                Amp.rotateAng(7)
                Amp.rotateAng({})
    
        def test_trim(self):
            # a new test for the trim module
            stlPath = os.path.abspath(os.getcwd()) + "\\tests\\sample_stl_sphere_BIN.stl"
            from AmpScan.core import AmpObject
            Amp = AmpObject(stlPath)
            with self.assertRaises(TypeError):
                Amp.planarTrim([], [])
            
    
    if __name__ == '__main__':
        unittest.main()