Recognize 0 in customInitialValues (Strilanc#461)

adgt · Oct 15, 2020 · 1f54b8e · 1f54b8e
1 parent 2978e99
commit 1f54b8e
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 1 deletion.
diff --git a/src/circuit/CircuitDefinition.js b/src/circuit/CircuitDefinition.js
@@ -153,7 +153,7 @@ class CircuitDefinition {
         if (newStateIndex !== undefined) {
             newVal = newStateIndex;
         }
-        if (newVal === undefined) {
+        if (newVal === undefined || newVal === 0) {
             m.delete(wire);
         } else {
             m.set(wire, newVal);

diff --git a/test/circuit/CircuitDefinition.test.js b/test/circuit/CircuitDefinition.test.js
@@ -1310,6 +1310,21 @@ suite.test("gateAtLocIsDisabledReason_multiwireOperations", () => {
 
 });
 
+suite.test("withSwitchedInitialStateOn", () => {
+    let c = circuit(`-
+                     -`);
+    assertThat(c.customInitialValues).isEqualTo(new Map());
+
+    c = c.withSwitchedInitialStateOn(0);
+    assertThat(c.customInitialValues).isEqualTo(new Map([[0, '1']]));
+
+    c = c.withSwitchedInitialStateOn(0);
+    assertThat(c.customInitialValues).isEqualTo(new Map([[0, '+']]));
+
+    c = c.withSwitchedInitialStateOn(0, 0);
+    assertThat(c.customInitialValues).isEqualTo(new Map());
+});
+
 suite.test("colCustomContextFromGates", () => {
     let c = circuit(`-A-B-
                      -A-A-