From 711ad900153aedb7b35896c0705c932972edded2 Mon Sep 17 00:00:00 2001
From: Basile Henry
Date: Fri, 27 Nov 2020 23:08:14 +0100
Subject: If can return a type

---
 dhall/src/operations/typecheck.rs                                   | 4 ++--
 .../type-inference/failure/unit/IfBranchesNotTermTypeOrKind.txt     | 4 ++--
 dhall/tests/type-inference/failure/unit/IfBranchesNotType.txt       | 6 ------
 3 files changed, 4 insertions(+), 10 deletions(-)
 delete mode 100644 dhall/tests/type-inference/failure/unit/IfBranchesNotType.txt

(limited to 'dhall')

diff --git a/dhall/src/operations/typecheck.rs b/dhall/src/operations/typecheck.rs
index 3016fea..bc0e864 100644
--- a/dhall/src/operations/typecheck.rs
+++ b/dhall/src/operations/typecheck.rs
@@ -350,8 +350,8 @@ pub fn typecheck_operation(
             if *x.ty().kind() != NirKind::from_builtin(Builtin::Bool) {
                 return span_err("InvalidPredicate");
             }
-            if y.ty().ty().as_const() != Some(Const::Type) {
-                return span_err("IfBranchMustBeTerm");
+            if y.ty().ty().as_const().is_none() {
+                return span_err("IfBranchMustBeTermTypeOrKind");
             }
             if y.ty() != z.ty() {
                 return span_err("IfBranchMismatch");
diff --git a/dhall/tests/type-inference/failure/unit/IfBranchesNotTermTypeOrKind.txt b/dhall/tests/type-inference/failure/unit/IfBranchesNotTermTypeOrKind.txt
index 8b46d1f..4023128 100644
--- a/dhall/tests/type-inference/failure/unit/IfBranchesNotTermTypeOrKind.txt
+++ b/dhall/tests/type-inference/failure/unit/IfBranchesNotTermTypeOrKind.txt
@@ -1,6 +1,6 @@
-Type error: error: IfBranchMustBeTerm
+Type error: error: IfBranchMustBeTermTypeOrKind
  --> <current file>:1:1
   |
 1 | if True then Kind else Kind
-  | ^^^^^^^^^^^^^^^^^^^^^^^^^^^ IfBranchMustBeTerm
+  | ^^^^^^^^^^^^^^^^^^^^^^^^^^^ IfBranchMustBeTermTypeOrKind
   |
diff --git a/dhall/tests/type-inference/failure/unit/IfBranchesNotType.txt b/dhall/tests/type-inference/failure/unit/IfBranchesNotType.txt
deleted file mode 100644
index 4c9ec28..0000000
--- a/dhall/tests/type-inference/failure/unit/IfBranchesNotType.txt
+++ /dev/null
@@ -1,6 +0,0 @@
-Type error: error: IfBranchMustBeTerm
- --> <current file>:1:1
-  |
-1 | if True then Type else Type
-  | ^^^^^^^^^^^^^^^^^^^^^^^^^^^ IfBranchMustBeTerm
-  |
-- 
cgit v1.2.3