From 19ae829bd19b4ffc09d57262b9ac6485ed5ed2b2 Mon Sep 17 00:00:00 2001
From: gg <ggordbegli@gmail.com>
Date: Wed, 12 Feb 2025 23:55:28 -0500
Subject: [PATCH] test float uop in sym_infer (#7456)

* float uop in sym_infer

* break line :(

* rerun mypy

* update GlobalCounters types

* revert type change and cast assignments to mem and ops

* cast inferred value to UOp in reshape

* cast hcq, update view reshape to handle inferred float

* rm extra space

* update error

* no type updates
---
 test/unit/test_uop_symbolic.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/test/unit/test_uop_symbolic.py b/test/unit/test_uop_symbolic.py
index d5fb8a5655..e4552679d6 100644
--- a/test/unit/test_uop_symbolic.py
+++ b/test/unit/test_uop_symbolic.py
@@ -617,6 +617,7 @@ class TestSymInfer(unittest.TestCase):
     c = Variable("c", 0, 10)
     var_vals = {a: 2, b: 3, c: 4}
     assert sym_infer(5, var_vals) == 5
+    assert sym_infer(4.2, var_vals) == 4.2
     assert sym_infer(a, var_vals) == 2
     assert sym_infer(b, var_vals) == 3
     assert sym_infer(a+b, var_vals) == 5