Improving/Expanding Functional Tests

[Benudek (benedikt herudek)]

submitted a PR for inspecting and asserting over fit and fit_one_cycle, hope it makes sense.

Test fitonecycle

All ears, if one can improve these tests here.

@sgugger