[add] ntrees to summary

natverse · May 28, 2021 · ad9f2ca · ad9f2ca
1 parent ffa4065
commit ad9f2ca
Show file tree

Hide file tree

Showing 5 changed files with 7 additions and 2 deletions.
diff --git a/R/summary.R b/R/summary.R
@@ -39,6 +39,8 @@ summary.neuronlist<-function(object, ..., include.attached.dataframe=FALSE) {
 #'   
 #'   \item{cable.length}
 #'   
+#'   \item{n.trees}
+#'   
 #'   }
 #' @export
 #' @examples 
@@ -50,7 +52,8 @@ summary.neuron<-function(object, ...) {
              segments=object$NumSegs,
              branchpoints=length(object$BranchPoints),
              endpoints=length(object$EndPoints),
-             cable.length=total_cable(object))
+             cable.length=total_cable(object),
+             n.trees=ifelse("nTrees" %in% names(object), object$nTrees, 1))
 }
 
 #' @export

diff --git a/man/summary.neuron.Rd b/man/summary.neuron.Rd
diff --git a/tests/testthat/test-summary.R b/tests/testthat/test-summary.R
@@ -2,7 +2,7 @@ context("summary")
 
 test_that("summary.neuronlist behaves", {
   expect_is(s <- summary(Cell07PNs), 'data.frame')
-  expect_equal_to_reference(s, file = 'testdata/summary_cell07pns.rda', tolerance=1e-3)
+  expect_equal_to_reference(s, file = 'testdata/summary_cell07pns.rds', tolerance=1e-3)
   expect_is(summary(kcs20, veclength=1.2), 'data.frame')
 })
 

diff --git a/tests/testthat/testdata/summary_cell07pns.rda b/tests/testthat/testdata/summary_cell07pns.rda
diff --git a/tests/testthat/testdata/summary_cell07pns.rds b/tests/testthat/testdata/summary_cell07pns.rds