InputGen test targets

Reviewed By: metascroy

Differential Revision: D57572670

fbshipit-source-id: ab39dab41ac14012781a6daf109f71716f0f9ef3

Author: manuelcandales

inputgen/argument/engine.py

@@ -28,7 +28,7 @@ def __init__(
         self.constraints = constraints
         self.deps = deps
         self.valid = valid
-        self.hierarchy = StructuralEngine.hierarchy(argtype)
+        self._hierarchy = StructuralEngine.hierarchy(argtype)
 
         self.gen_list_mode = set()
         for constraint in constraints:
@@ -54,7 +54,7 @@ def gen_structure_with_depth_and_length(
             yield ()
             return
 
-        attr = self.hierarchy[-(depth + 1)]
+        attr = self._hierarchy[-(depth + 1)]
 
         if attr in self.gen_list_mode:
             yield from self.gen_structure_with_depth(depth, focus, length)
@@ -81,7 +81,7 @@ def gen_structure_with_depth(
         length: Optional[int] = None,
         ix: Optional[int] = None,
     ):
-        attr = self.hierarchy[-(depth + 1)]
+        attr = self._hierarchy[-(depth + 1)]
 
         if ix is not None:
             args = (self.deps, length, ix)
@@ -104,7 +104,7 @@ def gen_structure_with_depth(
                 yield from self.gen_structure_with_depth_and_length(depth - 1, v, focus)
 
     def gen(self, focus: Attribute):
-        depth = len(self.hierarchy) - 1
+        depth = len(self._hierarchy) - 1
         yield from self.gen_structure_with_depth(depth, focus)
 
 

test/inputgen/test_meta_arg_engine.py

@@ -22,8 +22,9 @@ def test_tensor(self):
             cp.Value.Ne(lambda deps, dtype, struct: 0),
         ]
         deps = [2]
+        outarg = False
 
-        engine = MetaArgEngine(ArgType.Tensor, constraints, deps, True)
+        engine = MetaArgEngine(outarg, ArgType.Tensor, constraints, deps, True)
         ms = list(engine.gen(Attribute.DTYPE))
         self.assertEqual(len(ms), len(SUPPORTED_TENSOR_DTYPES))
         self.assertEqual({m.dtype for m in ms}, set(SUPPORTED_TENSOR_DTYPES))
@@ -50,14 +51,15 @@ def test_dim_list(self):
             ),
         ]
         deps = [2, 3]
+        outarg = False
 
-        engine = MetaArgEngine(ArgType.DimList, constraints, deps, True)
+        engine = MetaArgEngine(outarg, ArgType.DimList, constraints, deps, True)
         ms = list(engine.gen(Attribute.VALUE))
         self.assertEqual(len(ms), 1)
         self.assertTrue(1 <= len(ms[0].value) <= 5)
         self.assertTrue(all(v == 2 for v in ms[0].value))
 
-        engine = MetaArgEngine(ArgType.DimList, constraints, deps, False)
+        engine = MetaArgEngine(outarg, ArgType.DimList, constraints, deps, False)
         ms = list(engine.gen(Attribute.VALUE))
         self.assertEqual(len(ms), 1)
         self.assertTrue(1 <= len(ms[0].value) <= 5)