def test_sexmachine_bernoulliNB_load(self): s = DameSexmachine() m = s.bernoulliNB_load() predicted = m.predict([[ 0, 0, 1, 0, 21, 0, 0, 0, 0, 34, 2, 0, 0, 0, 0, 0, 0, 0, 0, 5, 0, 0, 0, 0, 0, 2, 0, 0, 0, 34, 1, 0, 1 ]]) n = np.array([2]) self.assertTrue(np.array_equal(predicted, n))
def test_sexmachine_bernoulliNB_load_method_returns_correct_result(self): s = DameSexmachine() m = s.bernoulliNB_load() predicted = m.predict([[ 0, 0, 1, 0, 21, 0, 0, 0, 0, 34, 2, 0, 0, 0, 0, 0, 0, 0, 0, 5, 0, 0, 0, 0, 0, 2, 0, 0, 0, 34, 1, 0, 1, 0]]) n = np.array([2]) self.assertTrue(np.array_equal(predicted, n))