Skip to content

Commit

Permalink
sample files10
Browse files Browse the repository at this point in the history
  • Loading branch information
Zamechek, Shawn committed Dec 16, 2016
1 parent 51a4d5b commit 37aecca
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions test/test_mfnbc.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,16 +13,16 @@
class TestMFNBC(unittest.TestCase):

def test_print_probs(self):
m = MFNBC('mfnbc/likeli_sample.csv', 'mfnbc/input_sample.csv', False)
m = MFNBC('likeli_sample.csv', 'input_sample.csv', False)
self.assertNotEqual(m.print_probs(), '')

def test_read_likelihoods(self):
d = MFNBC('mfnbc/likeli_sample.csv', 'mfnbc/input_sample.csv', False)
d = MFNBC('likeli_sample.csv', 'input_sample.csv', False)
d._read_likelihoods()
self.assertEqual(PROBS, PROBS)

def test_calc_denuminator(self):
m = MFNBC('mfnbc/likeli_sample.csv', 'mfnbc/input_sample.csv', False)
m = MFNBC('likeli_sample.csv', 'input_sample.csv', False)
m._read_likelihoods()
den = m._calc_denuminator(m.posteriors, m.probs['cat'], m.features)
self.assertEqual(den, 0.13666666666666666)
Expand Down

0 comments on commit 37aecca

Please sign in to comment.