'The dimension of the input should match the number of models unless
model_set_axis=False is given, in which case the input is copied across
all models.'

def test_scalar_parameters_2d_array_input(self):

    t = TModel_1_1([1, 2], [10, 20], n_models=2)
    y1 = t((np.arange(6).reshape(2, 3) * 100))
    assert (np.shape(y1) == (2, 3))
    assert np.all((y1 == [[11, 111, 211], [322, 422, 522]]))
    y2 = t((np.arange(6).reshape(2, 3) * 100), model_set_axis=False)
    assert (np.shape(y2) == (2, 2, 3))
    assert np.all((y2 == [[[11, 111, 211], [311, 411, 511]], [[22, 122, 222], [322, 422, 522]]]))
