Add static assertions to nodes (#6228)

astral-sh · Aug 1, 2023 · ecfdd8d · ecfdd8d
1 parent 07468f8
commit ecfdd8d
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 0 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/crates/ruff_python_ast/Cargo.toml b/crates/ruff_python_ast/Cargo.toml
@@ -28,6 +28,7 @@ once_cell = { workspace = true }
 rustc-hash = { workspace = true }
 serde = { workspace = true, optional = true }
 smallvec = { workspace = true }
+static_assertions = "1.1.0"
 
 [dev-dependencies]
 insta = { workspace = true }

diff --git a/crates/ruff_python_ast/src/nodes.rs b/crates/ruff_python_ast/src/nodes.rs
@@ -3044,3 +3044,16 @@ impl Ranged for crate::nodes::ArgWithDefault {
         self.range
     }
 }
+
+#[cfg(target_pointer_width = "64")]
+mod size_assertions {
+    #[allow(clippy::wildcard_imports)]
+    use super::*;
+    use static_assertions::assert_eq_size;
+
+    assert_eq_size!(Stmt, [u8; 168]);
+    assert_eq_size!(Expr, [u8; 80]);
+    assert_eq_size!(Constant, [u8; 32]);
+    assert_eq_size!(Pattern, [u8; 96]);
+    assert_eq_size!(Mod, [u8; 64]);
+}