Expand directed gnp test

Qiskit · Aug 1, 2023 · 361b128 · 361b128
1 parent 50e1e68
commit 361b128
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/tests/rustworkx_tests/test_random.py b/tests/rustworkx_tests/test_random.py
@@ -12,15 +12,17 @@
 
 import unittest
 import random
+from ddt import ddt, data
 
 import rustworkx
 
-
+@ddt
 class TestGNPRandomGraph(unittest.TestCase):
-    def test_random_gnp_directed(self):
-        graph = rustworkx.directed_gnp_random_graph(20, 0.5, seed=10)
-        self.assertEqual(len(graph), 20)
-        self.assertEqual(len(graph.edges()), 189)
+    @data((15, 20, 0.7, 156), (20, 10, 0.5, 189), (22, 6, 0.2, 91))
+    def test_random_gnp_directed(self, args):
+        graph = rustworkx.directed_gnp_random_graph(args[0], args[2], seed=args[1])
+        self.assertEqual(len(graph), args[0])
+        self.assertEqual(len(graph.edges()), args[3])
 
     def test_random_gnp_directed_empty_graph(self):
         graph = rustworkx.directed_gnp_random_graph(20, 0)