Spec tests for Pump

itu-square · Nov 9, 2022 · 236bf47 · 236bf47
1 parent 56f64f5
commit 236bf47
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/test/scala/symsim/examples/concrete/pumping/PumpSpec.scala b/src/test/scala/symsim/examples/concrete/pumping/PumpSpec.scala
@@ -39,8 +39,8 @@ class PumpSpec
     "The tank level never can be negative" in check {
       forAll (flow, head, head_mean, tank, time, water, past_head_mean, actions) {
         (f, h, hm, tl, t, w, phm, a) =>
-        val (s1, r) = Pump.step (PumpState (f, h, hm, tl, t, w, phm)) (a).head
-        s1.tl >= TANK_MIN
+        for (s1, r) <- Pump.step (PumpState (f, h, hm, tl, t, w, phm)) (a)
+        yield s1.tl >= TANK_MIN
       }
     }