add class catmaidneuron

* also test for it * closes #15
natverse · Apr 27, 2015 · 4bb78a3 · 4bb78a3
1 parent 022f6bc
commit 4bb78a3
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/R/catmaid_nat.R b/R/catmaid_nat.R
@@ -48,6 +48,7 @@ read.neuron.catmaid<-function(skid, pid=1L, conn=NULL, ...) {
   n[names(res[-1])]=res[-1]
   fields_to_include=c("url", "headers")
   n[fields_to_include]=attributes(res)[fields_to_include]
+  class(n)=c('catmaidneuron', 'neuron')
   n
 }
 

diff --git a/tests/testthat/test-fetch.R b/tests/testthat/test-fetch.R
@@ -39,6 +39,8 @@ test_that("get neuron", {
 test_that("read.neuron(s).catmaid", {
   if(!inherits(conn, 'try-error')){
     expect_is(n<-read.neuron.catmaid(pid=1, skid=10418394, conn=conn), 'neuron')
+    # check that we also have this specialised class
+    expect_is(n, 'catmaidneuron')
     expect_is(nl<-read.neurons.catmaid(c(4453485,10418394), conn=conn), 'neuronlist')
     expect_is(df<-as.data.frame(nl), 'data.frame')
     expect_is(df$name, "character")