tests

Signed-off-by: Justin Chu <[email protected]>

Author: justinchuby

src/onnx_ir/_convenience/_constructors_test.py

@@ -30,7 +30,7 @@ def test_tensor_handles_empty_sequence_with_dtype(self):
 class ValueConstructorTest(unittest.TestCase):
     def test_value_minimal_creation(self):
         """Test creating a value with just a name."""
-        value = _constructors.value("minimal")
+        value = _constructors.val("minimal")
 
         self.assertEqual(value.name, "minimal")
         self.assertIsNone(value.type)
@@ -39,7 +39,7 @@ def test_value_minimal_creation(self):
 
     def test_value_creation_with_sequence_shape(self):
         """Test that shape is correctly converted from sequence to Shape object."""
-        value = _constructors.value("test", ir.DataType.INT32, [1, 2, 3])
+        value = _constructors.val("test", ir.DataType.INT32, [1, 2, 3])
 
         self.assertEqual(value.name, "test")
         self.assertIsInstance(value.shape, ir.Shape)
@@ -48,7 +48,7 @@ def test_value_creation_with_sequence_shape(self):
     def test_value_creation_with_explicit_type(self):
         """Test value creation with explicit type parameter."""
         tensor_type = ir.TensorType(ir.DataType.DOUBLE)
-        value = _constructors.value("y", type=tensor_type, shape=[10])
+        value = _constructors.val("y", type=tensor_type, shape=[10])
 
         self.assertEqual(value.name, "y")
         self.assertEqual(value.type, tensor_type)
@@ -57,7 +57,7 @@ def test_value_creation_with_explicit_type(self):
     def test_value_creation_with_const_value(self):
         """Test value creation with const_value (initializer)."""
         const_tensor = ir.Tensor(np.array([1.0, 2.0, 3.0], dtype=np.float32), name="const")
-        value = _constructors.value("initializer", const_value=const_tensor)
+        value = _constructors.val("initializer", const_value=const_tensor)
 
         self.assertEqual(value.name, "initializer")
         self.assertEqual(value.type, ir.TensorType(ir.DataType.FLOAT))
@@ -66,7 +66,7 @@ def test_value_creation_with_const_value(self):
 
     def test_value_creation_with_dtype_only(self):
         """Test value creation with only dtype specified."""
-        value = _constructors.value("float_value", dtype=ir.DataType.FLOAT)
+        value = _constructors.val("float_value", dtype=ir.DataType.FLOAT)
 
         self.assertEqual(value.name, "float_value")
         self.assertEqual(value.type, ir.TensorType(ir.DataType.FLOAT))
@@ -78,25 +78,25 @@ def test_value_const_value_type_mismatch_error(self):
         wrong_type = ir.TensorType(ir.DataType.FLOAT)
 
         with self.assertRaisesRegex(ValueError, "The type does not match the const_value"):
-            _constructors.value("test", type=wrong_type, const_value=const_tensor)
+            _constructors.val("test", type=wrong_type, const_value=const_tensor)
 
     def test_value_const_value_dtype_mismatch_error(self):
         """Test that providing mismatched dtype with const_value raises ValueError."""
         const_tensor = ir.tensor([1.0, 2.0], dtype=ir.DataType.FLOAT)
 
         with self.assertRaisesRegex(ValueError, "The dtype does not match the const_value"):
-            _constructors.value("test", dtype=ir.DataType.INT32, const_value=const_tensor)
+            _constructors.val("test", dtype=ir.DataType.INT32, const_value=const_tensor)
 
     def test_value_const_value_shape_mismatch_error(self):
         """Test that providing mismatched shape with const_value raises ValueError."""
         const_tensor = ir.tensor([[1, 2], [3, 4]], dtype=ir.DataType.INT32)  # Shape: [2, 2]
 
         with self.assertRaisesRegex(ValueError, "The shape does not match the const_value"):
-            _constructors.value("test", shape=[3, 3], const_value=const_tensor)
+            _constructors.val("test", shape=[3, 3], const_value=const_tensor)
 
     def test_value_initialize_with_const_value(self):
         const_tensor = ir.tensor(np.array([[1.5, 2.5], [3.5, 4.5]], dtype=np.float64))
-        value = _constructors.value("test", const_value=const_tensor)
+        value = _constructors.val("test", const_value=const_tensor)
 
         self.assertEqual(value.name, "test")
         self.assertEqual(value.type, ir.TensorType(ir.DataType.DOUBLE))
@@ -105,14 +105,14 @@ def test_value_initialize_with_const_value(self):
 
     def test_value_creation_with_string_dimensions(self):
         """Test value creation with string dimensions in shape."""
-        value = _constructors.value("dynamic", ir.DataType.FLOAT, ["batch", "seq_len", 768])
+        value = _constructors.val("dynamic", ir.DataType.FLOAT, ["batch", "seq_len", 768])
 
         self.assertEqual(value.name, "dynamic")
         self.assertEqual(value.shape, ir.Shape(["batch", "seq_len", 768]))
 
     def test_value_creation_with_none_dimensions(self):
         """Test value creation with None dimensions in shape."""
-        value = _constructors.value("unknown", ir.DataType.INT64, [None, 10, None])
+        value = _constructors.val("unknown", ir.DataType.INT64, [None, 10, None])
 
         self.assertEqual(value.name, "unknown")
         self.assertEqual(value.shape, ir.Shape([None, 10, None]))