fixed solve_matrix

mdolab · Aug 16, 2018 · 0a38fd5 · 0a38fd5
1 parent dfc54a2
commit 0a38fd5
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 2 deletions.
diff --git a/openaerostruct/aerodynamics/solve_matrix.py b/openaerostruct/aerodynamics/solve_matrix.py
@@ -50,7 +50,6 @@ def solve_nonlinear(self, inputs, outputs):
 
     def linearize(self, inputs, outputs, partials):
         system_size = self.system_size
-
         self.lu = lu_factor(inputs['mtx'])
 
         partials['circulations', 'circulations'] = inputs['mtx'].flatten()

diff --git a/openaerostruct/aerodynamics/tests/test_solve_matrix.py b/openaerostruct/aerodynamics/tests/test_solve_matrix.py
@@ -1,4 +1,7 @@
 import unittest
+import numpy as np
+
+from openmdao.api import Group, IndepVarComp
 
 from openaerostruct.aerodynamics.solve_matrix import SolveMatrix
 from openaerostruct.utils.testing import run_test, get_default_surfaces
@@ -9,9 +12,26 @@ class Test(unittest.TestCase):
     def test(self):
         surfaces = get_default_surfaces()
 
+        group = Group()
         comp = SolveMatrix(surfaces=surfaces)
+
+        indep_var_comp = IndepVarComp()
+
+        system_size = 0
+        for surface in surfaces:
+            nx = surface['num_x']
+            ny = surface['num_y']
+            system_size += (nx - 1) * (ny - 1)
+
+
+        indep_var_comp.add_output('rhs', val=np.ones((system_size)), units='m/s')
+        indep_var_comp.add_output('mtx', val=np.identity((system_size)), units='1/m')
+
+        group.add_subsystem('indep_var_comp', indep_var_comp, promotes=['*'])
+        group.add_subsystem('solve_matrix', comp, promotes=['*'])
+
 
-        run_test(self, comp)
+        run_test(self, group)
 
 
 if __name__ == '__main__':