Convert turek example to nutils.units

Author: gertjanvanzwieten

examples/turek.py

@@ -1,7 +1,12 @@
 from nutils import cli, export, function, testing
 from nutils.mesh import gmsh
 from nutils.solver import System
-from nutils.SI import Length, Density, Viscosity, Velocity, Time, Pressure, Acceleration
+try:
+    from nutils.units.typing import Length, Density, Viscosity, Velocity, Time, Pressure, Acceleration
+except ModuleNotFoundError as e:
+    if hasattr(e, 'add_note'):
+        e.add_note("Consider installing the units package via: pip install nutils-units")
+    raise
 from nutils.expression_v2 import Namespace
 from collections import defaultdict, deque
 from dataclasses import dataclass
@@ -221,7 +226,7 @@ def main(domain: Domain = Domain(), solid: Optional[Solid] = Solid(), fluid: Opt
         if dynamic:
             ns.v, ns.a = dynamic.newmark_defo(ns.d)
         else:
-            ns.a = Acceleration.wrap(function.zeros((2,)))
+            ns.a = numpy.repeat(Acceleration.zero, 2)
 
         # Deformed geometry
         ns.x_i = 'xref_i + d_i'
@@ -260,8 +265,8 @@ def main(domain: Domain = Domain(), solid: Optional[Solid] = Solid(), fluid: Opt
     else: # fully rigid solid
 
         ns.x = ns.xref
-        ns.v = Velocity.wrap(function.zeros((2,)))
-        ns.a = Acceleration.wrap(function.zeros((2,)))
+        ns.v = numpy.repeat(Velocity.zero, 2)
+        ns.a = numpy.repeat(Acceleration.zero, 2)
 
     if fluid:
 
@@ -339,7 +344,7 @@ def main(domain: Domain = Domain(), solid: Optional[Solid] = Solid(), fluid: Opt
 
     DL = uxy = None # for unit tests only
 
-    for t in log.iter.fraction('timestep', dynamic.times) if dynamic else [Time.wrap(float('inf'))]:
+    for t in log.iter.fraction('timestep', dynamic.times) if dynamic else [Time.zero * float('inf')]:
 
         if dynamic:
             if solid: