def test_model_remove_component_argument_all(self): """Should remove an argument in an existing component.""" # Setup source_file = os.path.join(self.data_dir, 'DCMotor.mo') model = Model(source_file) # Act model.remove_component_argument('VsourceDC', 'DC', 'f') self.result = model.execute() # Assert self.assertHasAdditions(source_file, self.result, [' VsourceDC DC();']) self.assertHasDeletions(source_file, self.result, ['f=10'])
def test_model_remove_component_argument_last(self): """Should remove an argument in an existing component.""" # Setup source_file = os.path.join(self.data_dir, 'DCMotor.mo') model = Model(source_file) # Act model.remove_component_argument('ElectroMechanicalElement', 'EM', 'b') self.result = model.execute() # Assert self.assertHasAdditions(source_file, self.result, ['ElectroMechanicalElement EM(k=10, J=10 );']) self.assertHasDeletions(source_file, self.result, ['b=2'])