Coverage for docs/source/examples/vectorize.py: 100%

16 statements  

« prev     ^ index     » next       coverage.py v7.4.0, created at 2024-01-07 22:13 -0500

1"Vectorization demonstration" 

2from gpkit import Model, Variable, Vectorize 

3 

4class Test(Model): 

5 """A simple scalar model 

6 

7 Upper Unbounded 

8 --------------- 

9 x 

10 """ 

11 def setup(self): 

12 x = self.x = Variable("x") 

13 return [x >= 1] 

14 

15print("SCALAR") 

16m = Test() 

17m.cost = m["x"] 

18print(m.solve(verbosity=0).summary()) 

19 

20print("__________\n") 

21print("VECTORIZED") 

22with Vectorize(3): 

23 m = Test() 

24m.cost = m["x"].prod() 

25m.append(m["x"][1] >= 2) 

26print(m.solve(verbosity=0).summary())