survival 0.0.3 → 0.0.4

data/lib/survival/KaplanMeier.rb

@@ -46,9 +46,9 @@ class KaplanMeier
 
       #The median survival time is calculated as the smallest survival time 
       #for which the survivor function is less than or equal to 0.5
-      if ( (s_t <= 50.0) && (median.nil?) ){
+      if ( (s_t <= 50.0) && (median.nil?) )
         median = t
-      }
+      end
 
       #Add to plot points
       if include_horizontal

data/lib/survival/version.rb

@@ -1,3 +1,3 @@
 module Survival
-  VERSION = "0.0.3"
+  VERSION = "0.0.4"
 end

data/test/survival_test.rb

@@ -6,7 +6,7 @@ class SurvivalTest < ActiveSupport::TestCase
 
   test 'km plot' do
     p = KaplanMeier.generate_plot_points(SurvivalSample.create_survival_objects(@@survivals))
-    assert_equal [[0, 100], [3, 100], [3, 100.0], [4, 100.0], [4, 0.0]], p
+    assert_equal [[[0, 100], [3, 100], [3, 100.0], [4, 100.0], [4, 0.0]], 4], p
   end
 
   test 'survival_sample array' do

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: survival
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.0.4
   prerelease: 
 platform: ruby
 authors: