Nothing
context("MILP: model-api")
describe("nvars()", {
it("returns the number of variables", {
model <- MILPModel() %>%
add_variable(x[i], i = 1:10, type = "binary") %>%
add_variable(y[i], i = 1:5, type = "continuous") %>%
add_variable(z[i], i = 1:2, type = "integer")
result <- nvars(model)
expect_equivalent(10, result$binary)
expect_equivalent(5, result$continuous)
expect_equivalent(2, result$integer)
expect_null(names(result$continuous))
expect_null(names(result$binary))
expect_null(names(result$integer))
})
it("returns 0 for a model without variables", {
model <- MILPModel()
result <- nvars(model)
expect_equivalent(0, result$binary)
expect_equivalent(0, result$continuous)
expect_equivalent(0, result$integer)
})
})
describe("objective_function()", {
it("returns a list with a vector and a constant by default", {
model <- MILPModel() %>%
add_variable(x[i], i = 1:9) %>%
set_objective(sum_expr(i * x[i], i = 1:9) + 10)
result <- objective_function(model)
expect_equal(c(1:9), as.vector(result$solution))
expect_equal(10, result$constant)
})
it("returns handles models without objective function", {
model <- MILPModel() %>%
add_variable(x[i], i = 1:10)
result <- objective_function(model)
expect_equal(rep.int(0, 10), as.vector(result$solution))
expect_equal(0, result$constant)
})
it("returns a sparse vector", {
n <- 2
model <- MILPModel() %>%
add_variable(x[i, j],
i = 1:n, j = 1:n,
type = "integer", lb = 0, ub = 1
) %>%
set_objective(sum_expr(x[i, j], i = 1:n, j = 1:n)) %>%
add_constraint(sum_expr(x[i, j], i = 1:n, j = 1:n) <= 10)
result <- objective_function(model)
expected <- Matrix::sparseVector(
x = rep.int(1L, n^2),
i = seq_len(n^2), length = n^2
)
expect_equal(result$solution, expected)
})
it("returns an all 0 vector if objective function is numeric", {
model <- MILPModel() %>%
add_variable(x[i], i = 1:10) %>%
set_objective(1)
result <- objective_function(model)
expect_equal(length(result$solution), 10L)
expect_equal(sum(result$solution), 0)
expect_equal(result$constant, 1)
})
})
describe("variable_keys()", {
it("returns a vector of unique variable keys", {
model <- MILPModel() %>%
add_variable(x[i], i = 1:3)
result <- variable_keys(model)
expect_equal(c("x[1]", "x[2]", "x[3]"), result)
})
it("works with more than 1 index var", {
model <- MILPModel() %>%
add_variable(x[i, j], i = 1:2, j = 1:2)
result <- variable_keys(model)
expect_equal(c("x[1,1]", "x[2,1]", "x[1,2]", "x[2,2]"), result)
})
it("works with vars without an index", {
model <- MILPModel() %>%
add_variable(x[i, j], i = 1, j = 1) %>%
add_variable(a)
result <- variable_keys(model)
expect_equal(c("a", "x[1,1]"), result)
})
it("sorts keys alphabetically", {
model <- MILPModel() %>%
add_variable(x[i], i = 1:3) %>%
add_variable(y[i], i = 1:3)
result <- variable_keys(model)
expect_equal(sort(result), result)
})
it("returns an empty character vector if model has no vars", {
expect_equal(character(0), variable_keys(MILPModel()))
})
})
describe("nconstraints()", {
it("returns the number of constraints of a model", {
m <- MILPModel() %>%
add_variable(x[i], i = 1:5) %>%
add_constraint(x[i] <= 10, i = 1:5) %>%
add_constraint(x[i] >= 5, i = 1:5)
expect_equal(nconstraints(m), 10)
})
})
describe("extract_constraints()", {
it("returns a list of named elements", {
model <- MILPModel() %>%
add_variable(x[i], i = 1:3) %>%
add_variable(y[i], i = 1:3) %>%
add_constraint(x[i] + y[i] <= 1, i = 1:3)
result <- extract_constraints(model)
expect_true(is.list(result))
expect_true(all(c("matrix", "rhs", "sense") %in% names(result)))
})
it("returns the constraint matrix as a matrix", {
model <- MILPModel() %>%
add_variable(x[i], i = 1:3) %>%
add_variable(y[i], i = 1:3) %>%
add_constraint(x[i] + y[i] <= 1, i = 1:3)
result <- extract_constraints(model)
exp_matrix <- matrix(c(
1, 0, 0, 0, 1, 0, 0, 0, 1,
1, 0, 0, 0, 1, 0, 0, 0, 1
), ncol = 6, nrow = 3)
expect_equivalent(exp_matrix, as.matrix(result$matrix))
})
it("returns the constraint right hand side", {
model <- MILPModel() %>%
add_variable(x[i], i = 1:3) %>%
add_variable(y[i], i = 1:3) %>%
add_constraint(x[i] + y[i] <= 1, i = 1:3)
result <- extract_constraints(model)
expect_equal(c(1, 1, 1), result$rhs)
})
it("returns the constraint sense", {
model <- MILPModel() %>%
add_variable(x[i], i = 1:3) %>%
add_variable(y[i], i = 1:3) %>%
add_constraint(x[i] + y[i] <= 1, i = 1:3)
result <- extract_constraints(model)
expect_equal(c("<=", "<=", "<="), result$sense)
})
it("returns a sparse Matrix with column-oriented encoding", {
model <- MILPModel() %>%
add_variable(x[i], i = 1:3) %>%
add_variable(y[i], i = 1:3) %>%
add_constraint(x[i] + y[i] <= 1, i = 1:3)
result <- extract_constraints(model)
expect_s4_class(result$matrix, "dgCMatrix")
model <- MILPModel() %>%
add_variable(x[i], i = 1:3) %>%
add_constraint(x[i] <= 0, i = 1:3)
result <- extract_constraints(model)
expect_s4_class(result$matrix, "dgCMatrix")
expect_equal(result$rhs, rep.int(0, 3))
expect_equivalent(
as.matrix(result$matrix),
matrix(c(
1, 0, 0,
0, 1, 0,
0, 0, 1
), ncol = 3)
)
})
it("works with non indexed variables", {
model <- MILPModel() %>%
add_variable(x) %>%
add_constraint(x <= 1)
result <- extract_constraints(model)
expect_equivalent(matrix(1, nrow = 1, ncol = 1), as.matrix(result$matrix))
})
it("supports underscores in variables", {
# bug #115 20170217
model <- MILPModel() %>%
add_variable(x_a[i], i = 1:3) %>%
set_objective(sum_expr(x_a[i], i = 1:3)) %>%
add_constraint(x_a[1] == 1)
expect_equal(1, sum(extract_constraints(model)$matrix))
})
it("does not emit 0 coefficients", {
model <- MILPModel() %>%
add_variable(x[i], i = 1:3) %>%
add_constraint(x[i] <= 1 + x[c(2, 2, 2)], i = 1:3)
result <- extract_constraints(model)
expect_equal(result$rhs, c(1, 1, 1))
expect_equivalent(as.matrix(result$matrix), t(matrix(c(
1, -1, 0,
0, 0, 0,
0, -1, 1
), ncol = 3, nrow = 3)))
expect_true(all(result$matrix@x != 0))
})
})
describe("variable_types()", {
it("returns the variable types in the correct order", {
model <- MILPModel() %>%
add_variable(x, type = "binary") %>%
add_variable(y, type = "continuous") %>%
add_variable(z, type = "integer")
result <- variable_types(model)
expect_equal(factor(c("binary", "continuous", "integer")), result)
})
it("returns the variable types in the correct order for index variables", {
model <- MILPModel() %>%
add_variable(x[i], type = "integer", i = 1:2) %>%
add_variable(a[i], type = "binary", i = 1:2)
expected <- factor(c("binary", "binary", "integer", "integer"))
expect_equal(expected, variable_types(model))
})
it("returns an empty vector if model has no variables", {
expect_equal(factor(), variable_types(MILPModel()))
})
})
describe("variable_bounds()", {
it("returns a list with the correct variable bounds", {
model <- MILPModel() %>%
add_variable(x, type = "binary") %>%
add_variable(y, type = "continuous", lb = 2) %>%
add_variable(z, type = "integer", ub = 3)
result <- variable_bounds(model)
expected <- list(
lower = c(0, 2, -Inf),
upper = c(1, Inf, 3)
)
expect_equal(expected, result)
})
it("works with indexed variables", {
model <- MILPModel() %>%
add_variable(x[i], i = 1:3, lb = 1, ub = 3, type = "integer")
result <- variable_bounds(model)
expected <- list(
lower = c(1, 1, 1),
upper = c(3, 3, 3)
)
expect_equal(expected, result)
})
it("returns a list with empty numerics of model has no variables", {
expect_equal(list(
lower = numeric(0),
upper = numeric(0)
), variable_bounds(MILPModel()))
})
it("returns the bounds in the order of the constraint matrix", {
n <- 2
model <- MILPModel() %>%
add_variable(x[i, j], i = 1:n, j = 1:n, type = "binary") %>%
add_variable(u[i], i = 1:n, lb = 1, ub = n) %>%
set_objective(0) %>%
add_constraint(u[i] + 1 <= u[j] + n * (1 - x[i, j]), i = 1:n, j = 1:n)
result <- variable_bounds(model)
expect_equal(c(1, 1, 0, 0, 0, 0), result$lower)
expect_equal(c(2, 2, 1, 1, 1, 1), result$upper)
})
})
test_that("bug 20170312: variable_keys has wrong ordering", {
model <- MILPModel() %>%
add_variable(x[i, j],
i = 1:2, j = 1:3, type = "integer",
lb = 0, ub = 5
) %>%
set_bounds(x[i, i], i = 1:2, lb = 1, ub = 1)
result <- variable_bounds(model)
keys <- variable_keys(model)
expect_equal(c(1, 0, 0, 1, 0, 0), result$lower)
expect_equal(c(1, 5, 5, 1, 5, 5), result$upper)
})
Any scripts or data that you put into this service are public.
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.