Updated VectorDist print method

Author: Raphael

NEWS.md

@@ -21,6 +21,7 @@
 ## Patches
 
 - Added wrapper for VectorDistribution for quick concatenation of constructed distributions, `c.Distribution`
+- `VectorDistribution` `print` method more in line with base R vectors
 
 
 # distr6 1.3.0

R/Wrapper_VectorDistribution.R

@@ -257,6 +257,14 @@ VectorDistribution$set("public","modelTable", function(){
   private$.wrappedModels
 })
 
+VectorDistribution$set("public", "strprint", function(n = 100){
+  names <- as.character(self$modelTable()$shortname)
+  lng <- length(names)
+  if(lng >(2*n))
+    names = c(names[1:n], "...", names[(lng-n+1):lng])
+
+  return(names)
+})
 VectorDistribution$set("public", "getParameterValue", function(...){
   message("Vector Distribution should not be used to get/set parameters. Try to use '[' first.")
   return(NULL)

tests/testthat/test-VectorDistribution.R

@@ -138,3 +138,12 @@ test_that("shared d/p/q/r",{
                                                 Binom3 = qbinom(0.42,6,0.2)))
 })
 
+test_that("print",{
+  a = VectorDistribution$new(distribution = "Binomial", params = list(list(prob = 0.1, size = 2),
+                                                                      list(prob = 0.6, size = 4),
+                                                                      list(prob = 0.2, size = 6)))
+  expect_equal(a$strprint(), c("Binom1","Binom2","Binom3"))
+  expect_equal(a$strprint(1), c("Binom1","...","Binom3"))
+  expect_output(a$print(), "Binom1")
+})
+