Move test.rs from test to src