Fix random choice ofv coins

MassimoLauria · Sep 30, 2020 · db790e3 · db790e3
1 parent 1a5fd66
commit db790e3
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 2 deletions.
diff --git a/cnfgen/clihelpers/simple_helpers.py b/cnfgen/clihelpers/simple_helpers.py
@@ -173,10 +173,9 @@ def build_cnf(args):
         n = args.n
         if args.plant:
             planted = {}
-            signs = random.choices([True, False], k=n)
 
             for i in range(n):
-                planted["x_{0}".format(i + 1)] = signs[i]
+                planted["x_{0}".format(i + 1)] = random.choice([True, False])
 
             return RandomKCNF(args.k,
                               args.n,

diff --git a/tests/test_randomcnf.py b/tests/test_randomcnf.py
@@ -65,6 +65,21 @@ def test_random_cnf_planted():
         assert len(c) == 3
 
 
+def test_random_cnf_planted_2():
+    planted = [{
+        "x_1": True,
+        "x_2": False,
+        "x_3": True,
+        "x_4": False,
+        "x_5": True,
+        "x_6": True
+    }]
+    F = RandomKCNF(3, 6, 10, planted_assignments=planted)
+    for c in F:
+        assert len(c) == 3
+    assert len(F) == 10
+
+
 def test_randomkcnf_cnfgen():
     cnfgen(["cnfgen", 'randkcnf', 4, 6, 10])