Adds tests for move_allocation