def grenoble_estimation(main_data_dir):
    with open(main_data_dir+'grenoble.yml') as model_file,\
            open(main_data_dir+'grenoble.csv') as data_file:
        model = choice_model.MultinomialLogit.from_yaml(model_file)
        model.load_data(data_file)
    interface = choice_model.PylogitInterface(model)
    interface.estimate()
    return interface
def simple_multinomial_pylogit_estimation(simple_multinomial_model_with_data):
    interface = choice_model.PylogitInterface(
        simple_multinomial_model_with_data)
    interface.estimate()
    return interface
def simple_multinomial_pylogit_interface(simple_multinomial_model_with_data):
    return choice_model.PylogitInterface(simple_multinomial_model_with_data)
 def test_no_data(self, simple_multinomial_model):
     with pytest.raises(choice_model.interface.interface.NoDataLoaded):
         choice_model.PylogitInterface(simple_multinomial_model)
 def test_simple_model(self, simple_model):
     with pytest.raises(TypeError):
         choice_model.PylogitInterface(simple_model)
 def test_multinomial_logit(self, simple_multinomial_model_with_data):
     interface = choice_model.PylogitInterface(
             simple_multinomial_model_with_data)
     assert interface.model == simple_multinomial_model_with_data