now it is two tests, one with an integer, asserting grad is 0
one with a float, asserting grad is what it was before