From 236bf4779bf046cc33d8ee42a09dd89eb2a01ee3 Mon Sep 17 00:00:00 2001
From: mohsen-ghaffari1992 <mohg@itu.dk>
Date: Mon, 31 Oct 2022 15:07:20 +0100
Subject: [PATCH] Spec tests for Pump

---
 .../scala/symsim/examples/concrete/pumping/PumpSpec.scala     | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/test/scala/symsim/examples/concrete/pumping/PumpSpec.scala b/src/test/scala/symsim/examples/concrete/pumping/PumpSpec.scala
index c7836603..1cd7ccd7 100644
--- 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
       }
     }