Source code for pyNastran.op2.test.op2_unit_tests

import os
import unittest
import pyNastran
test_path = pyNastran.__path__[0]

from pyNastran.op2.op2 import OP2
from pyNastran.op2.test.test_op2 import run_op2
from pyNastran.bdf.test.bdf_unit_tests import Tester


class TestOP2(Tester):
    def _spike(self):
        op2 = OP2()
        op2.set_results('solidStress.oxx')
        op2.read_op2(op2_filename, vectorized=False)

    def test_set_results(self):
        folder = os.path.abspath(os.path.join(test_path, '..', 'models'))
        op2_filename = os.path.join(folder, 'solid_bending', 'solid_bending.op2')
        op2 = OP2()
        op2.set_results('stress')
        op2.read_op2(op2_filename, vectorized=False)
        self.assertEqual(len(op2.cpenta_stress), 0), len(op2.cpenta_stress)
        self.assertEqual(len(op2.chexa_stress), 0), len(op2.chexa_stress)
        self.assertEqual(len(op2.ctetra_stress), 1), len(op2.ctetra_stress)
        self.assertEqual(len(op2.displacements), 0), len(op2.displacements)

        op2 = OP2()
        op2.set_results(['stress', 'displacements'])
        op2.read_op2(op2_filename, vectorized=False)
        self.assertEqual(len(op2.cpenta_stress), 0), len(op2.cpenta_stress)
        self.assertEqual(len(op2.chexa_stress), 0), len(op2.chexa_stress)
        self.assertEqual(len(op2.ctetra_stress), 1), len(op2.ctetra_stress)
        self.assertEqual(len(op2.displacements), 1), len(op2.displacements)

        op2 = OP2()
        op2.set_results('stress')
        op2.read_op2(op2_filename, vectorized=True)
        self.assertEqual(len(op2.cpenta_stress), 0), len(op2.cpenta_stress)
        self.assertEqual(len(op2.chexa_stress), 0), len(op2.chexa_stress)
        self.assertEqual(len(op2.ctetra_stress), 1), len(op2.ctetra_stress)
        self.assertEqual(len(op2.displacements), 0), len(op2.displacements)

        op2 = OP2()
        op2.set_results(['stress', 'displacements'])
        op2.read_op2(op2_filename, vectorized=True)
        self.assertEqual(len(op2.cpenta_stress), 0), len(op2.cpenta_stress)
        self.assertEqual(len(op2.chexa_stress), 0), len(op2.chexa_stress)
        self.assertEqual(len(op2.ctetra_stress), 1), len(op2.ctetra_stress)
        self.assertEqual(len(op2.displacements), 1), len(op2.displacements)

    def test_op2_01(self):
        op2_filename = os.path.join('solid_bending', 'solid_bending.op2')
        folder = os.path.abspath(os.path.join(test_path, '..', 'models'))
        make_geom = False
        write_bdf = False
        write_f06 = True
        debug = False
        op2file = os.path.join(folder, op2_filename)
        run_op2(op2file, make_geom=make_geom, write_bdf=write_bdf, iSubcases=[],
                write_f06=write_f06, is_vector=False,
                debug=debug, stopOnFailure=True, binary_debug=True)
        assert os.path.exists('debug.out'), os.listdir('.')
        os.remove('debug.out')

        make_geom = False
        write_bdf = False
        write_f06 = True
        run_op2(op2file, make_geom=make_geom, write_bdf=write_bdf, iSubcases=[],
                write_f06=write_f06, is_vector=True,
                debug=debug, stopOnFailure=True, binary_debug=True)
        assert os.path.exists('debug.out'), os.listdir('.')
        os.remove('debug.out')

    def test_op2_02(self):
        op2_filename = os.path.join('plate_py', 'plate_py.op2')
        folder = os.path.abspath(os.path.join(test_path, '..', 'models'))
        make_geom = False
        write_bdf = False
        write_f06 = False
        debug = False
        op2file = os.path.join(folder, op2_filename)
        run_op2(op2file, make_geom=make_geom, write_bdf=write_bdf, iSubcases=[],
                write_f06=write_f06, is_vector=False,
                debug=debug, stopOnFailure=True)

        make_geom = False
        write_bdf = False
        write_f06 = True
        run_op2(op2file, make_geom=make_geom, write_bdf=write_bdf, iSubcases=[],
                write_f06=write_f06, is_vector=True,
                debug=debug, stopOnFailure=True)

if __name__ == '__main__':  # pragma: no cover
    unittest.main()