Add tests for allocations