I'm not sure that we need 2 equal tests. I think you should add new parameter `bool` to the first test that means whether `nonzero_rows` is `0` or something else