forked from OpenMDAO/Aviary
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request OpenMDAO#590 from xjjiang/test_mass_builders
Test mass builders
- Loading branch information
Showing
4 changed files
with
124 additions
and
7 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,52 @@ | ||
| import unittest | ||
|
|
||
| import numpy as np | ||
| import openmdao.api as om | ||
|
|
||
| from openmdao.utils.assert_utils import assert_check_partials, assert_near_equal | ||
|
|
||
| from aviary.subsystems.mass.mass_builder import CoreMassBuilder | ||
| from aviary.variable_info.enums import LegacyCode | ||
| from aviary.variable_info.variables import Aircraft | ||
| from aviary.variable_info.variable_meta_data import _MetaData as BaseMetaData | ||
| import aviary.api as av | ||
|
|
||
| FLOPS = LegacyCode.FLOPS | ||
|
|
||
|
|
||
| class TestFLOPSMassBuilder(av.TestSubsystemBuilderBase): | ||
| """ | ||
| That class inherits from TestSubsystemBuilder. So all the test functions are | ||
| within that inherited class. The setUp() method prepares the class and is run | ||
| before the test methods; then the test methods are run. | ||
| """ | ||
|
|
||
| def setUp(self): | ||
| self.subsystem_builder = CoreMassBuilder( | ||
| 'test_core_mass', meta_data=BaseMetaData, code_origin=FLOPS) | ||
| self.aviary_values = av.AviaryValues() | ||
| self.aviary_values.set_val(Aircraft.Design.USE_ALT_MASS, False, units='unitless') | ||
| self.aviary_values.set_val(Aircraft.Engine.NUM_ENGINES, [1], units='unitless') | ||
| self.aviary_values.set_val( | ||
| Aircraft.Engine.NUM_WING_ENGINES, [2], units='unitless') | ||
|
|
||
|
|
||
| class TestFLOPSMassBuilderAltMass(av.TestSubsystemBuilderBase): | ||
| """ | ||
| That class inherits from TestSubsystemBuilder. So all the test functions are | ||
| within that inherited class. The setUp() method prepares the class and is run | ||
| before the test methods; then the test methods are run. | ||
| """ | ||
|
|
||
| def setUp(self): | ||
| self.subsystem_builder = CoreMassBuilder( | ||
| 'test_core_mass', meta_data=BaseMetaData, code_origin=FLOPS) | ||
| self.aviary_values = av.AviaryValues() | ||
| self.aviary_values.set_val(Aircraft.Design.USE_ALT_MASS, True, units='unitless') | ||
| self.aviary_values.set_val(Aircraft.Engine.NUM_ENGINES, [1], units='unitless') | ||
| self.aviary_values.set_val( | ||
| Aircraft.Engine.NUM_WING_ENGINES, [2], units='unitless') | ||
|
|
||
|
|
||
| if __name__ == '__main__': | ||
| unittest.main() |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,63 @@ | ||
| import unittest | ||
|
|
||
| import numpy as np | ||
| import openmdao.api as om | ||
|
|
||
| from openmdao.utils.assert_utils import assert_check_partials, assert_near_equal | ||
|
|
||
| from aviary.subsystems.mass.mass_builder import CoreMassBuilder | ||
|
|
||
| from aviary.variable_info.enums import LegacyCode | ||
| from aviary.variable_info.variables import Aircraft | ||
| from aviary.variable_info.variable_meta_data import _MetaData as BaseMetaData | ||
| import aviary.api as av | ||
|
|
||
| GASP = LegacyCode.GASP | ||
|
|
||
|
|
||
| class TestGASPMassBuilderHybrid(av.TestSubsystemBuilderBase): | ||
| """ | ||
| That class inherits from TestSubsystemBuilder. So all the test functions are | ||
| within that inherited class. The setUp() method prepares the class and is run | ||
| before the test methods; then the test methods are run. | ||
| """ | ||
|
|
||
| def setUp(self): | ||
| self.subsystem_builder = CoreMassBuilder( | ||
| 'test_core_mass', meta_data=BaseMetaData, code_origin=GASP) | ||
| self.aviary_values = av.AviaryValues() | ||
| self.aviary_values.set_val( | ||
| Aircraft.Design.PART25_STRUCTURAL_CATEGORY, 3, units='unitless') | ||
| self.aviary_values.set_val( | ||
| Aircraft.Propulsion.TOTAL_NUM_ENGINES, 2, units='unitless') | ||
| self.aviary_values.set_val( | ||
| Aircraft.Electrical.HAS_HYBRID_SYSTEM, True, units='unitless') | ||
| self.aviary_values.set_val(Aircraft.Wing.HAS_FOLD, True, units='unitless') | ||
| self.aviary_values.set_val(Aircraft.Wing.HAS_STRUT, True, units='unitless') | ||
| self.aviary_values.set_val(Aircraft.Engine.NUM_ENGINES, [1], units='unitless') | ||
|
|
||
|
|
||
| class TestGASPMassBuilder(av.TestSubsystemBuilderBase): | ||
| """ | ||
| That class inherits from TestSubsystemBuilder. So all the test functions are | ||
| within that inherited class. The setUp() method prepares the class and is run | ||
| before the test methods; then the test methods are run. | ||
| """ | ||
|
|
||
| def setUp(self): | ||
| self.subsystem_builder = CoreMassBuilder( | ||
| 'test_core_mass', meta_data=BaseMetaData, code_origin=GASP) | ||
| self.aviary_values = av.AviaryValues() | ||
| self.aviary_values.set_val( | ||
| Aircraft.Design.PART25_STRUCTURAL_CATEGORY, 3, units='unitless') | ||
| self.aviary_values.set_val( | ||
| Aircraft.Propulsion.TOTAL_NUM_ENGINES, 2, units='unitless') | ||
| self.aviary_values.set_val( | ||
| Aircraft.Electrical.HAS_HYBRID_SYSTEM, False, units='unitless') | ||
| self.aviary_values.set_val(Aircraft.Wing.HAS_FOLD, True, units='unitless') | ||
| self.aviary_values.set_val(Aircraft.Wing.HAS_STRUT, True, units='unitless') | ||
| self.aviary_values.set_val(Aircraft.Engine.NUM_ENGINES, [1], units='unitless') | ||
|
|
||
|
|
||
| if __name__ == '__main__': | ||
| unittest.main() |