update test_NB() method