Moved tests to test file
Updated allocateTest
Forgot to update actual test files for imports