Handle scalars in monadic ≠ and ∪

1 files changed, 10 insertions, 0 deletions

diff --git a/functions.c b/functions.c
index ed0fb36..2425798 100644
--- a/functions.c
+++ b/functions.c
@@ -369,6 +369,13 @@ fnSame(Array *right)
 Array *
 fnUniqueMask(Array *right)
 {
+	if(right->rank == 0){
+		Array *res = allocarray(AtypeInt, 1, 1);
+		res->shape[0] = 1;
+		res->intdata[0] = 1;
+		return res;
+	}
+
 	return rundfn(L"(⍳≢⍵)≤⊃∘⍸¨↓≡⌾⍨⌷∘⍵¨⍳≢⍵", nil, nil, nil, right);
 }
 
@@ -620,6 +627,9 @@ fnEnlist(Array *right)
 Array *
 fnUnique(Array *right)
 {
+	if(right->rank == 0)
+		return fnRavel(right);
+
 	return rundfn(L"(≠⍵)⌿⍵", nil, nil, nil, right);
 }