[internal] ComparableExpr (f)strings and bytes made invariant under concatenation (#13301)

This commit is contained in:
Dylan 2024-09-25 09:58:57 -05:00 committed by GitHub
parent ca0ae0a484
commit f27a8b8c7a
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 173 additions and 38 deletions

View file

@ -0,0 +1,47 @@
use ruff_python_ast::comparable::ComparableExpr;
use ruff_python_parser::{parse_expression, ParseError};
#[test]
fn concatenated_strings_compare_equal() -> Result<(), ParseError> {
let split_contents = r#"'a' 'b' r'\n raw'"#;
let value_contents = r#"'ab\\n raw'"#;
let split_parsed = parse_expression(split_contents)?;
let value_parsed = parse_expression(value_contents)?;
let split_compr = ComparableExpr::from(split_parsed.expr());
let value_compr = ComparableExpr::from(value_parsed.expr());
assert_eq!(split_compr, value_compr);
Ok(())
}
#[test]
fn concatenated_bytes_compare_equal() -> Result<(), ParseError> {
let split_contents = r#"b'a' b'b'"#;
let value_contents = r#"b'ab'"#;
let split_parsed = parse_expression(split_contents)?;
let value_parsed = parse_expression(value_contents)?;
let split_compr = ComparableExpr::from(split_parsed.expr());
let value_compr = ComparableExpr::from(value_parsed.expr());
assert_eq!(split_compr, value_compr);
Ok(())
}
#[test]
fn concatenated_fstrings_compare_equal() -> Result<(), ParseError> {
let split_contents = r#"f"{foo!r} this" r"\n raw" f" and {bar!s} that""#;
let value_contents = r#"f"{foo!r} this\\n raw and {bar!s} that""#;
let split_parsed = parse_expression(split_contents)?;
let value_parsed = parse_expression(value_contents)?;
let split_compr = ComparableExpr::from(split_parsed.expr());
let value_compr = ComparableExpr::from(value_parsed.expr());
assert_eq!(split_compr, value_compr);
Ok(())
}