Move test to corresponding files

[JErnestoMtz]

Now must of the tests are in lib.rs, I think we should move each test to the corresponding file it is testing.