MilesCranmer commited on
Commit
bfb135a
1 Parent(s): 9335949

Add constraints test with custom operator

Browse files
Files changed (1) hide show
  1. test/test.py +3 -0
test/test.py CHANGED
@@ -50,6 +50,9 @@ class TestPipeline(unittest.TestCase):
50
  verbosity=0,
51
  **self.default_test_kwargs,
52
  procs=0,
 
 
 
53
  )
54
  model.fit(self.X, y)
55
  equations = model.equations
 
50
  verbosity=0,
51
  **self.default_test_kwargs,
52
  procs=0,
53
+ # Test custom operators with constraints:
54
+ nested_constraints={"square_op": {"square_op": 3}},
55
+ constraints={"square_op": 10},
56
  )
57
  model.fit(self.X, y)
58
  equations = model.equations