fix ternary test

johnynek · Nov 20, 2024 · 28f9ebd · 28f9ebd
1 parent 20c099e
commit 28f9ebd
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/cli/src/test/scala/org/bykn/bosatsu/codegen/python/CodeTest.scala b/cli/src/test/scala/org/bykn/bosatsu/codegen/python/CodeTest.scala
@@ -521,7 +521,17 @@ else:
             assert(tern == f.simplify)
           }
         case whoKnows =>
-          assert(tern == Code.Ternary(t.simplify, whoKnows, f.simplify))
+          if (tern == whoKnows) {
+            (t.simplify, f.simplify) match {
+              case (Code.Const.One | Code.Const.True, Code.Const.False | Code.Const.Zero) =>
+                ()
+              case tf =>
+                fail(s"$tern == $whoKnows but (t,f) = $tf")
+            }
+          }
+          else {
+            assert(tern == Code.Ternary(t.simplify, whoKnows, f.simplify))
+          }
       }
     }
   }