func TestGetSegmentMLModel(t *testing.T) {
httpmock.Activate()
defer httpmock.DeactivateAndReset()
mock.RegisterSegmentMLMocks()
client := NewLytics(mock.MockApiKey, nil, nil)
model, err := client.GetSegmentMLModel(mock.MockSegmentMLID)
assert.Equal(t, err, nil)
assert.Equal(t, model.Name, mock.MockSegmentMLID)
assert.Equal(t, model.Importance["lytics_score_frequency"], 0.5850666666666667)
assert.Equal(t, model.Summary.Conf.FalseNegative, 3)
}
func TestGetSegmentMLModels(t *testing.T) {
httpmock.Activate()
defer httpmock.DeactivateAndReset()
mock.RegisterSegmentMLMocks()
client := NewLytics(mock.MockApiKey, nil, nil)
models, err := client.GetSegmentMLModels()
assert.Equal(t, err, nil)
assert.Equal(t, len(models), 2)
first, ok := models[mock.MockSegmentMLID]
assert.T(t, ok)
second, ok := models["goal_anonymous::goal_known"]
assert.T(t, ok)
assert.Equal(t, first.Importance["lytics_score_recency"], 0.10316666666666663)
assert.Equal(t, first.Summary.Mse, 0.27024152611111113)
assert.Equal(t, second.Name, "goal_anonymous::goal_known")
assert.Equal(t, second.FieldTypes["goal_known"], "numeric")
}
