re-write test_naive_bayes2