Updated tests

Author: JWock82

Testing/test_2D_frames.py

@@ -23,7 +23,7 @@ def tearDown(self):
         sys.stdout = sys.__stdout__
 
     def test_XY_gravity_load(self):
-        
+
         # A First Course in the Finite Element Method, 4th Edition
         # Daryl L. Logan
         # Problem 5.30
@@ -119,7 +119,7 @@ def test_XY_member_ptload(self):
         # Update the expected value to an appropriate precision
         expected_RZ = 0.00022794540510395617
         self.assertAlmostEqual(calculated_RZ/expected_RZ, 1.0, 2)
-    
+
     def test_YZ_gravity_load(self):
         # A First Course in the Finite Element Method, 4th Edition
         # Daryl L. Logan
@@ -177,7 +177,7 @@ def test_YZ_gravity_load(self):
                 self.assertAlmostEqual(node.RxnFZ['Combo 1']/values['RxnFZ'], 1.0, 2)
                 self.assertAlmostEqual(node.RxnFY['Combo 1']/values['RxnFY'], 1.0, 2)
                 self.assertAlmostEqual(node.RxnMX['Combo 1']/values['RxnMX'], 1.0, 2)
-        
+
         # Check displacements at N3 and N4
         correct_displacements = [('N3', {'DY': -6.666757,
                                          'RX':  0.032}),
@@ -226,7 +226,7 @@ def test_XZ_ptload(self):
                 # Two decimal place accuracy requires +/-0.5% accuracy
                 # one decimal place requires +/-5%
                 self.assertAlmostEqual(calculated_reaction/rxn, 1.0, 2)
-    
+
     def test_Kassimali_3_35(self):
         """
         Tests against Kassimali example 3.35.
@@ -239,7 +239,7 @@ def test_Kassimali_3_35(self):
         """
 
         frame = FEModel3D()
-        
+
         frame.add_node('A', 0, 0, 0)
         frame.add_node('B', 0, 0, 24)
         frame.add_node('C', 12, 0, 0)

Testing/test_beam_rotation.py

@@ -39,3 +39,6 @@ def test_beam_rotation():
     assert Mz_min == pt.approx(-17.677, rel=1e-2)
     assert My_max == pt.approx(17.677, rel=1e-2)
     assert My_min == pt.approx(0.0, rel=1e-2)
+
+if __name__ == '__main__':
+    test_beam_rotation()

Testing/test_continuous_beam.py

@@ -47,10 +47,10 @@ def test_continuous_beam_moments():
 
 #     assert isinstance(model.members['M1'].plot_axial('1.0D', 100), matplotlib.figure.Figure)
 #     plt.close()
-    
+
 #     assert isinstance(model.members['M1'].plot_deflection('dy', '1.0D', 100), matplotlib.figure.Figure)
 #     plt.close()
 
 if __name__ == '__main__':
-    test_continuous_beam_moments()
-    # test_plots()
+    # test_continuous_beam_moments()
+    # test_plots()

Testing/test_reactions.py

@@ -1,39 +1,20 @@
 # -*- coding: utf-8 -*-
-"""
-MIT License
 
-Copyright (c) 2020 D. Craig Brinck, SE; tamalone1
-"""
-
-import unittest
 from Pynite import FEModel3D
-import math
-import sys
-from io import StringIO
-
-class Test_2D_Frame(unittest.TestCase):
-    """Tests for reaction results
 
-    :param unittest: _description_
-    :type unittest: _type_
-    """
 
-    def setUp(self):
-        # Suppress printed output temporarily
-        sys.stdout = StringIO()
+def test_spring_reactions():
 
-    def tearDown(self):
-        # Reset the print function to normal
-        sys.stdout = sys.__stdout__
+    model = FEModel3D()
 
-    def test_spring_reactions(self):
+    model.add_node('N1', 0, 0, 0)
+    model.add_node_load('N1', 'FY', -5)
+    model.def_support('N1', True, False, True, True, True, True)
+    model.def_support_spring('N1', 'DY', '5', '-')
+    model.analyze()
 
-        model = FEModel3D()
+    assert model.nodes['N1'].RxnFY['Combo 1'] == 5, 'Failed spring reaction test.'
 
-        model.add_node('N1', 0, 0, 0)
-        model.add_node_load('N1', 'FY', -5)
-        model.def_support('N1', True, False, True, True, True, True)
-        model.def_support_spring('N1', 'DY', '5', '-')
-        model.analyze()
 
-        self.assertEqual(model.nodes['N1'].RxnFY['Combo 1'], 5)
+if __name__ == '__main__':
+    test_spring_reactions()