示例#1
0
    def test_generic_calls(self):
        """
        check whether all generic simulation and postprocessing modules are registered using the general getter call
        """
        # ----------------------
        # simulation

        # solver
        self.assertEqual([(ODEInt, "ODEInt")],
                         get_registered_modules(SimulationModule, Solver))

        # trajectory generators
        self.assertEqual([(SmoothTransition, "SmoothTransition"),
                          (HarmonicTrajectory, "HarmonicTrajectory"),
                          (Setpoint, "Setpoint")
                          ],
                         get_registered_modules(SimulationModule, Trajectory))

        # controllers
        self.assertEqual([(PIDController, "PIDController")],
                         get_registered_modules(SimulationModule, Controller))

        # feedforward
        self.assertEqual([(PyTrajectory, "PyTrajectory")],
                         get_registered_modules(SimulationModule, Feedforward))

        # mixers
        self.assertEqual([(AdditiveMixer, "AdditiveMixer")],
                         get_registered_modules(SimulationModule, ModelMixer))
        self.assertEqual([(AdditiveMixer, "AdditiveMixer")],
                         get_registered_modules(SimulationModule, ObserverMixer))

        # limiter
        self.assertEqual([(ModelInputLimiter, "ModelInputLimiter")],
                         get_registered_modules(SimulationModule, Limiter))

        # sensors
        self.assertEqual([(DeadTimeSensor, "DeadTimeSensor")],
                         get_registered_modules(SimulationModule, Sensor))

        # disturbance
        self.assertEqual([(GaussianNoise, "GaussianNoise")],
                         get_registered_modules(SimulationModule, Disturbance))

        # ----------------------
        # processing

        # post-processors
        self.assertEqual([(StepResponse, "StepResponse"),
                          (PlotAll, "PlotAll")
                          ],
                         get_registered_modules(ProcessingModule, PostProcessingModule))
示例#2
0
    def test_generic_calls(self):
        """
        check whether all generic simulation and postprocessing modules are registered using the general getter call
        """
        # ----------------------
        # simulation

        # solver
        self.assertEqual([(ODEInt, "ODEInt")],
                         get_registered_modules(SimulationModule, Solver))

        # trajectory generators
        self.assertEqual([(SmoothTransition, "SmoothTransition"),
                          (HarmonicTrajectory, "HarmonicTrajectory"),
                          (Setpoint, "Setpoint")],
                         get_registered_modules(SimulationModule, Trajectory))

        # controllers
        self.assertEqual([(PIDController, "PIDController")],
                         get_registered_modules(SimulationModule, Controller))

        # feedforward
        self.assertEqual([(PyTrajectory, "PyTrajectory")],
                         get_registered_modules(SimulationModule, Feedforward))

        # mixers
        self.assertEqual([(AdditiveMixer, "AdditiveMixer")],
                         get_registered_modules(SimulationModule, ModelMixer))
        self.assertEqual([(AdditiveMixer, "AdditiveMixer")],
                         get_registered_modules(SimulationModule,
                                                ObserverMixer))

        # limiter
        self.assertEqual([(ModelInputLimiter, "ModelInputLimiter")],
                         get_registered_modules(SimulationModule, Limiter))

        # sensors
        self.assertEqual([(DeadTimeSensor, "DeadTimeSensor")],
                         get_registered_modules(SimulationModule, Sensor))

        # disturbance
        self.assertEqual([(GaussianNoise, "GaussianNoise")],
                         get_registered_modules(SimulationModule, Disturbance))

        # ----------------------
        # processing

        # post-processors
        self.assertEqual([(StepResponse, "StepResponse"),
                          (PlotAll, "PlotAll")],
                         get_registered_modules(ProcessingModule,
                                                PostProcessingModule))
示例#3
0
 def test_string_call(self):
     """
     the calls should also work with names instead of class objects
     """
     self.assertEqual([(ODEInt, "ODEInt")],
                      get_registered_modules("SimulationModule", "Solver"))
     self.assertEqual(
         [(StepResponse, "StepResponse"), (PlotAll, "PlotAll")],
         get_registered_processing_modules("PostProcessingModule"))
示例#4
0
 def test_string_call(self):
     """
     the calls should also work with names instead of class objects
     """
     self.assertEqual([(ODEInt, "ODEInt")],
                      get_registered_modules("SimulationModule", "Solver"))
     self.assertEqual([(StepResponse, "StepResponse"),
                       (PlotAll, "PlotAll")
                       ],
                      get_registered_processing_modules("PostProcessingModule"))