Add unit tests for Block