Add unit tests for the `initialization/1` built-in directive