Added support for code blocks as assert_same argument