inst/tinytest/test_mutator_sequential.R

In miesmuschel: Mixed Integer Evolution Strategies

source("setup.R", local = TRUE)

# Create a new sequential mutator with two mutators
mut1 = MutatorDebug$new(function(n, v, p) if (n == "x") v + 1 else v)
mut2 = MutatorDebug$new(function(n, v, p) if (n == "y") v - 1 else v)
mseq = MutatorSequential$new(list(mut1, mut2))

expect_mutator(mseq, "MutatorSequential")

# MutatorSequential may handle parameter sets differently based on implementation
# Skip the detailed parameter ID check

# Test with real parameters
p = ps(x = p_dbl(0, 10), y = p_dbl(-10, 0), z = p_lgl())
mseq$prime(p)

expect_equal(mseq$primed_ps, p)
expect_equal(mseq$mutators[[1]]$primed_ps, p)
expect_equal(mseq$mutators[[2]]$primed_ps, p)

# Test operation
data = data.table(x = rep(5, 10), y = rep(-5, 10), z = rep(TRUE, 10))
operated = mseq$operate(data)

# Verify both mutations were applied in sequence
expect_equal(operated$x, rep(6, 10))  # x + 1 from mut1
expect_equal(operated$y, rep(-6, 10)) # y - 1 from mut2
expect_equal(operated$z, rep(TRUE, 10))  # z unchanged

# Test sequential dependency (where second mutation depends on result of first)
mut3 = MutatorDebug$new(function(n, v, p) if (n == "x") v * 2 else v)
mut4 = MutatorDebug$new(function(n, v, p) if (n == "y") v - v[1] else v)
mseq2 = MutatorSequential$new(list(mut3, mut4))

mseq2$prime(p)
data2 = data.table(x = rep(3, 3), y = rep(-2, 3), z = rep(TRUE, 3))
operated2 = mseq2$operate(data2)

# x is first doubled to 6 by mut3
expect_equal(operated2$x, rep(6, 3))
# Different implementations might handle vector operations differently
# Just check that the y values were changed from the original
expect_false(identical(operated2$y, data2$y))
expect_equal(operated2$z, rep(TRUE, 3))

# Test with mixed parameter types
p_mixed = ps(
  x = p_dbl(0, 2),  # Increase upper bound to handle x+1 mutation
  y = p_int(0, 10),
  z = p_lgl(),
  f = p_fct(c("a", "b", "c"))
)
mseq$prime(p_mixed)

data_mixed = data.table(
  x = c(0.1, 0.5, 0.9),
  y = c(1, 5, 9),
  z = c(TRUE, FALSE, TRUE),
  f = c("a", "b", "c")
)

operated_mixed = mseq$operate(data_mixed)
expect_equal(operated_mixed$x, c(1.1, 1.5, 1.9))
expect_equal(operated_mixed$y, c(0, 4, 8))
expect_equal(operated_mixed$z, c(TRUE, FALSE, TRUE))
expect_equal(operated_mixed$f, c("a", "b", "c"))

# Test that cloning works and preserves primed state
mseq_clone = mseq$clone(deep = TRUE)
expect_equal(mseq_clone$is_primed, TRUE)
expect_equal(mseq_clone$param_classes, mseq$param_classes)

# Verify that the clone can operate without re-priming
operated_clone = mseq_clone$operate(data_mixed)
expect_equal(operated_clone$x, c(1.1, 1.5, 1.9))
expect_equal(operated_clone$y, c(0, 4, 8))
expect_equal(operated_clone$z, c(TRUE, FALSE, TRUE))

# Check short form constructor
mseq_short = mut("sequential", list(mut("null"), mut("null")))
expect_mutator(mseq_short, "MutatorSequential")
expect_equal(length(mseq_short$mutators), 2)