add unit tests for initialize

[mzywang]

• #231
• #230 👈
• #229
• #228
• #227
• main

This stack of pull requests is managed by Graphite. Learn more about stacking.

Join @mzywang and the rest of your teammates on Graphite