Type check body of with statements
Extract _assert_expression_is_type_checked in tests