fixes test imports

jwdebelius · Jan 19, 2016 · 560dbdc · 560dbdc
1 parent f74ae60
commit 560dbdc
Showing 1 changed file with 11 additions and 11 deletions.
diff --git a/tests/test_per_sample.py b/tests/test_per_sample.py
@@ -235,7 +235,7 @@ def test_plot_alpha(self):
 
         (tgroup, tgalpha, tsalpha, txlabel) = \
             agps._plot_alpha(sample=sample,
-                             alpha_map=smap_,
+                             alpha_map=map_,
                              alpha_field='alpha',
                              debug=True)
 
@@ -261,9 +261,9 @@ def test_plot_alpha_no_sample(self):
             )
         sample = 'AlP0'
         kvalue = 'alpha does not have an alpha diversity value for AlP0.'
-        tvalue = plot_alpha(sample=sample,
-                            alpha_map=self.map_,
-                            alpha_field='alpha')
+        tvalue = agps._plot_alpha(sample=sample,
+                                  alpha_map=map_,
+                                  alpha_field='alpha')
         self.assertEqual(tvalue, kvalue)
 
     def test_plot_alpha_alpha_field_error(self):
@@ -282,9 +282,9 @@ def test_plot_alpha_alpha_field_error(self):
                      'HOST_SUBJECT_ID', 'NATIONALITY', 'alpha'],
             )
         with self.assertRaises(ValueError):
-            plot_alpha(sample='VeP0',
-                       alpha_map=map_,
-                       alpha_field='InCryptid')
+            agps._plot_alpha(sample='VeP0',
+                             alpha_map=map_,
+                             alpha_field='InCryptid')
 
     def test_plot_alpha_group_field_error(self):
         map_ = pd.DataFrame(
@@ -302,10 +302,10 @@ def test_plot_alpha_group_field_error(self):
                      'HOST_SUBJECT_ID', 'NATIONALITY', 'alpha'],
             )
         with self.assertRaises(ValueError):
-            plot_alpha(sample='VeP0',
-                       alpha_map=map_,
-                       alpha_field='alpha',
-                       group_field='BODY_HABITAT')
+            agps._plot_alpha(sample='VeP0',
+                             alpha_map=map_,
+                             alpha_field='alpha',
+                             group_field='BODY_HABITAT')
 
 if __name__ == '__main__':
     main()