Source code for moe_examples.tests.next_point_via_simple_endpoint_test

# -*- coding: utf-8 -*-
"""Integration test for next_point_via_simple_endpoint MOE example."""
from moe.optimal_learning.python.constant import TEST_OPTIMIZER_MULTISTARTS, TEST_OPTIMIZER_NUM_RANDOM_SAMPLES, TEST_GRADIENT_DESCENT_PARAMETERS

from moe_examples.tests.moe_example_test_case import MoeExampleTestCase
from moe_examples.next_point_via_simple_endpoint import run_example


[docs]class TestNextPointsViaSimpleEndpoint(MoeExampleTestCase):

    """Test the next_point_via_simple_endpoint MOE example."""

[docs]    def test_example_runs(self):
        """Simple integration test for example."""
        run_example(
                num_points_to_sample=1,
                verbose=False,
                testapp=self.testapp,
                optimizer_info={
                    'num_multistarts': TEST_OPTIMIZER_MULTISTARTS,
                    'num_random_samples': TEST_OPTIMIZER_NUM_RANDOM_SAMPLES,
                    'optimizer_parameters': TEST_GRADIENT_DESCENT_PARAMETERS._asdict(),
                    },
                rest_port=1337,
        )