torax.stepper.tests package

Submodules

torax.stepper.tests.pydantic_model_test module

class torax.stepper.tests.pydantic_model_test.PydanticModelTest(*args, **kwargs)[source]

Bases: TestCase

test_build_stepper_from_config_linear(stepper_model=<class 'torax.stepper.pydantic_model.LinearThetaMethod'>, expected_type=<class 'torax.stepper.linear_theta_method.LinearThetaMethod'>): Builds a stepper from the config.

test_build_stepper_from_config_newton_raphson(stepper_model=<class 'torax.stepper.pydantic_model.NewtonRaphsonThetaMethod'>, expected_type=<class 'torax.stepper.nonlinear_theta_method.NewtonRaphsonThetaMethod'>): Builds a stepper from the config.

test_build_stepper_from_config_optimizer(stepper_model=<class 'torax.stepper.pydantic_model.OptimizerThetaMethod'>, expected_type=<class 'torax.stepper.nonlinear_theta_method.OptimizerThetaMethod'>): Builds a stepper from the config.

Module contents

Tests for the TORAX stepper module.