tests/testthat/test-toJSON-complex.R

In jsonlite: A Simple and Robust JSON Parser and Generator for R

test_that("Encoding Complex", {
  expect_equal(toJSON(complex(real=2, imaginary=2)), "[\"2+2i\"]");
  expect_equal(toJSON(complex(real=NA, imaginary=2)), "[\"NA\"]");
  expect_equal(toJSON(complex(real=1, imaginary=NA)), "[\"NA\"]");
  expect_equal(toJSON(complex(real=NA, imaginary=2), na="null"), "[null]");
});

test_that("Encoding Complex in Data Frame", {
  expect_equal(toJSON(data.frame(foo=complex(real=1, imaginary=2))), "[{\"foo\":\"1+2i\"}]");
  expect_equal(toJSON(data.frame(foo=complex(real=NA, imaginary=2))), "[{}]");
  expect_equal(toJSON(data.frame(foo=complex(real=NA, imaginary=2)), na="string"), "[{\"foo\":\"NA\"}]");
  expect_equal(toJSON(data.frame(foo=complex(real=NA, imaginary=2)), na="null"), "[{\"foo\":null}]");
});

test_that("Encoding Complex as list", {
  x <- complex(real=c(1,2,NA), imaginary=3:1);
  expect_equal(toJSON(x), "[\"1+3i\",\"2+2i\",\"NA\"]");
  expect_equal(toJSON(x, complex="list"), "{\"real\":[1,2,\"NA\"],\"imaginary\":[3,2,1]}");
  expect_equal(toJSON(data.frame(foo=x), complex="list"), "[{\"foo\":{\"real\":1,\"imaginary\":3}},{\"foo\":{\"real\":2,\"imaginary\":2}},{\"foo\":{\"imaginary\":1}}]");
  expect_equal(toJSON(data.frame(foo=x), complex="list", na="string"), "[{\"foo\":{\"real\":1,\"imaginary\":3}},{\"foo\":{\"real\":2,\"imaginary\":2}},{\"foo\":{\"real\":\"NA\",\"imaginary\":1}}]");
  expect_equal(toJSON(data.frame(foo=x), complex="list", dataframe="columns"), "{\"foo\":{\"real\":[1,2,\"NA\"],\"imaginary\":[3,2,1]}}");
});