mirrors/ruff
1
0
Fork 0
mirror of https://github.com/astral-sh/ruff.git synced 2025-08-03 10:22:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

[red-knot] Infer float and complex literal expressions (#12689)

Browse source 
## Summary

This PR implements type inference for float and complex literal
expressions.

## Test Plan

Add test cases for both types.
This commit is contained in:
Dhruv Manilawala 2024-08-06 11:54:28 +05:30 • committed by GitHub
parent 39dd732e27
commit 5cc3fed9a8
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 18 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

23
crates/red_knot_python_semantic/src/types/infer.rs
View file

@ -933,8 +933,10 @@ impl<'db> TypeInferenceBuilder<'db> {
.as_i64()
.map(Type::IntLiteral)
.unwrap_or_else(|| builtins_symbol_ty_by_name(self.db, "int").instance()),
// TODO float or complex
_ => Type::Unknown,
ast::Number::Float(_) => builtins_symbol_ty_by_name(self.db, "float").instance(),
ast::Number::Complex { .. } => {
builtins_symbol_ty_by_name(self.db, "complex").instance()
}
}
}
@ -1641,12 +1643,23 @@ mod tests {
}
#[test]
fn resolve_literal() -> anyhow::Result<()> {
fn number_literal() -> anyhow::Result<()> {
let mut db = setup_db();
db.write_file("src/a.py", "x = 1")?;
db.write_dedented(
"src/a.py",
"
a = 1
b = 9223372036854775808
c = 1.45
d = 2j
",
)?;
assert_public_ty(&db, "src/a.py", "x", "Literal[1]");
assert_public_ty(&db, "src/a.py", "a", "Literal[1]");
assert_public_ty(&db, "src/a.py", "b", "int");
assert_public_ty(&db, "src/a.py", "c", "float");
assert_public_ty(&db, "src/a.py", "d", "complex");
Ok(())
}