tests/testthat/test-serialize.R

In asl/rssa: A Collection of Methods for Singular Spectrum Analysis

library(testthat);
library(Rssa);
source(system.file("extdata", "common.test.methods.R", package = "Rssa"));
context("Serialization");

test_that("Serialization works correctly", {
  Ls <- c(17, 100, 222, 234);
  kinds <- c("1d-ssa", "toeplitz-ssa");
  svd.methods <- c("eigen", "nutrlan", "propack", "svd");
  groups <- list(1, 1:2, 3:5, 1:5);
  len <- 100;
  neig = 15;

  for (kind in kinds) for (svd.method in svd.methods) {
    if (identical(kind, "toeplitz-ssa") && identical(svd.method, "svd"))
      next;

    for (L in Ls) {
      suppressWarnings(ss <- ssa(co2, L = L, kind = kind, svd.method = svd.method, neig = neig, force.decompose = TRUE));

      # Serialize ssa-object to raw vector
      rw <- serialize(ss, connection = NULL);

      # Unserialize ssa-object
      ss.uns <- unserialize(rw);

      expect_equal(ss.uns$U, ss$U);
      expect_equal(ss.uns$V, ss$V);
      expect_equal(ss.uns$lambda, ss$lambda);

      expect_equal(reconstruct(ss.uns, groups = groups), reconstruct(ss, groups = groups));
      expect_equal(rforecast(ss.uns, groups = groups, len = len, base = "original"),
                   rforecast(ss, groups = groups, len = len, base = "original"));

      expect_equal(rforecast(ss.uns, groups = groups, len = len, base = "reconstructed"),
                   rforecast(ss, groups = groups, len = len, base = "reconstructed"));

      expect_equal(vforecast(ss.uns, groups = groups, len = len),
                   vforecast(ss, groups = groups, len = len));
    }
  }
});

test_that("Serialization works correctly for 2d SSA", {
  N <- c(110, 117);

  set.seed(1);
  field <- matrix(rnorm(prod(N)), N[1], N[2]);

  Ls <- list(c(17, 16), c(50, 56));
  svd.methods <- c("nutrlan", "propack");
  groups <- list(1, 1:2, 3:5, 1:5, 1:10);
  neig = 15;

  for (svd.method in svd.methods) {
    for (L in Ls) {
      set.seed(1);
      ss <- ssa(field, L = L, kind = "2d-ssa", svd.method = svd.method, neig = neig, force.decompose = TRUE);

      # Serialize ssa-object to raw vector
      rw <- serialize(ss, connection = NULL);

      # Unserialize ssa-object
      ss.uns <- unserialize(rw);

      expect_equal(ss.uns$U, ss$U);
      expect_equal(ss.uns$V, ss$V);
      expect_equal(ss.uns$lambda, ss$lambda);

      expect_equal(reconstruct(ss.uns, groups = groups), reconstruct(ss, groups = groups));
    }
  }
});