added two new test cases for the assign_needy_zeros method