From 05daf01e6760a682ba0d7f42f84bc15d030f7252 Mon Sep 17 00:00:00 2001
From: Florian Weik <fweik@icp.uni-stuttgart.de>
Date: Thu, 5 Dec 2019 20:39:12 +0100
Subject: [PATCH] testsuite: dpd: Script momentum conservation check for dpd +
 langevin

---
 testsuite/python/dpd.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/testsuite/python/dpd.py b/testsuite/python/dpd.py
index 5be601023cd..28f0c8c87fa 100644
--- a/testsuite/python/dpd.py
+++ b/testsuite/python/dpd.py
@@ -73,7 +73,7 @@ def check_total_zero(self):
 
     def single(self, with_langevin=False):
         """Test velocity distribution of a dpd fluid with a single type."""
-        N = 250
+        N = 500
         s = self.s
         s.part.add(pos=s.box_l * np.random.random((N, 3)))
         kT = 2.3
@@ -95,7 +95,9 @@ def single(self, with_langevin=False):
         error_tol = 0.01
         self.check_velocity_distribution(
             v_stored, v_minmax, bins, error_tol, kT)
-        self.check_total_zero()
+
+        if not with_langevin:
+            self.check_total_zero()
 
     def test_single(self):
         self.single()