import control.tests.statefbk_test as statefbk_test

import control.tests.stochsys_test as stochsys_test

import control.tests.trdata_test as trdata_test

import control.tests.timeplot_test as timeplot_test

@pytest.mark.parametrize("module, prefix", [

(control, ""), (control.flatsys, "flatsys."), (control.optimal, "optimal.")

# @parametrize messes up the check, but we know it is there

pass

elif source and source.find('unrecognized keyword') < 0:

elif source and source.find('unrecognized keyword') < 0 and \

source.find('unexpected keyword') < 0:

warnings.warn(

f"'unrecognized keyword' not found in unit test "

f"for {name}")

function(*args, **kwargs, unknown=None)

@pytest.mark.parametrize(

"function", [control.time_response_plot, control.TimeResponseData.plot])

def test_time_response_plot_kwargs(function):

# Create a system for testing

response = control.step_response(control.rss(4, 2, 2))

# Call the plotting function normally and make sure it works

function(response)

# Now add an unrecognized keyword and make sure there is an error

with pytest.raises(AttributeError,

match="(has no property|unexpected keyword)"):

function(response, unknown=None)

#

# List of all unit tests that check for unrecognized keywords

#

'gangof4_plot': test_matplotlib_kwargs,

'input_output_response': test_unrecognized_kwargs,

'interconnect': interconnect_test.test_interconnect_exceptions,

'time_response_plot': timeplot_test.test_errors,

'linearize': test_unrecognized_kwargs,

'lqe': test_unrecognized_kwargs,

'lqr': test_unrecognized_kwargs,

'StateSpace.__init__': test_unrecognized_kwargs,

'StateSpace.sample': test_unrecognized_kwargs,

'TimeResponseData.__call__': trdata_test.test_response_copy,

'TimeResponseData.plot': timeplot_test.test_errors,

'TransferFunction.__init__': test_unrecognized_kwargs,

'TransferFunction.sample': test_unrecognized_kwargs,

'optimal.OptimalControlProblem.__init__':