1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
|
package hm
import (
"testing"
"github.com/stretchr/testify/assert"
)
func TestFunctionTypeBasics(t *testing.T) {
fnType := NewFnType(TypeVariable('a'), TypeVariable('a'), TypeVariable('a'))
if fnType.Name() != "→" {
t.Errorf("FunctionType should have \"→\" as a name. Got %q instead", fnType.Name())
}
if fnType.String() != "a → a → a" {
t.Errorf("Expected \"a → a → a\". Got %q instead", fnType.String())
}
if !fnType.Arg().Eq(TypeVariable('a')) {
t.Error("Expected arg of function to be 'a'")
}
if !fnType.Ret(false).Eq(NewFnType(TypeVariable('a'), TypeVariable('a'))) {
t.Error("Expected ret(false) to be a → a")
}
if !fnType.Ret(true).Eq(TypeVariable('a')) {
t.Error("Expected final return type to be 'a'")
}
// a very simple fn
fnType = NewFnType(TypeVariable('a'), TypeVariable('a'))
if !fnType.Ret(true).Eq(TypeVariable('a')) {
t.Error("Expected final return type to be 'a'")
}
ftv := fnType.FreeTypeVar()
if len(ftv) != 1 {
t.Errorf("Expected only one free type var")
}
for _, fas := range fnApplyTests {
fn := fas.fn.Apply(fas.sub).(*FunctionType)
if !fn.Eq(fas.expected) {
t.Errorf("Expected %v. Got %v instead", fas.expected, fn)
}
}
// bad shit
f := func() {
NewFnType(TypeVariable('a'))
}
assert.Panics(t, f)
}
var fnApplyTests = []struct {
fn *FunctionType
sub Subs
expected *FunctionType
}{
{NewFnType(TypeVariable('a'), TypeVariable('a')), mSubs{'a': proton, 'b': neutron}, NewFnType(proton, proton)},
{NewFnType(TypeVariable('a'), TypeVariable('b')), mSubs{'a': proton, 'b': neutron}, NewFnType(proton, neutron)},
{NewFnType(TypeVariable('a'), TypeVariable('b')), mSubs{'c': proton, 'd': neutron}, NewFnType(TypeVariable('a'), TypeVariable('b'))},
{NewFnType(TypeVariable('a'), TypeVariable('b')), mSubs{'a': proton, 'c': neutron}, NewFnType(proton, TypeVariable('b'))},
{NewFnType(TypeVariable('a'), TypeVariable('b')), mSubs{'c': proton, 'b': neutron}, NewFnType(TypeVariable('a'), neutron)},
{NewFnType(electron, proton), mSubs{'a': proton, 'b': neutron}, NewFnType(electron, proton)},
// a -> (b -> c)
{NewFnType(TypeVariable('a'), TypeVariable('b'), TypeVariable('a')), mSubs{'a': proton, 'b': neutron}, NewFnType(proton, neutron, proton)},
{NewFnType(TypeVariable('a'), TypeVariable('a'), TypeVariable('b')), mSubs{'a': proton, 'b': neutron}, NewFnType(proton, proton, neutron)},
{NewFnType(TypeVariable('a'), TypeVariable('b'), TypeVariable('c')), mSubs{'a': proton, 'b': neutron}, NewFnType(proton, neutron, TypeVariable('c'))},
{NewFnType(TypeVariable('a'), TypeVariable('c'), TypeVariable('b')), mSubs{'a': proton, 'b': neutron}, NewFnType(proton, TypeVariable('c'), neutron)},
// (a -> b) -> c
{NewFnType(NewFnType(TypeVariable('a'), TypeVariable('b')), TypeVariable('a')), mSubs{'a': proton, 'b': neutron}, NewFnType(NewFnType(proton, neutron), proton)},
}
func TestFunctionType_FlatTypes(t *testing.T) {
fnType := NewFnType(TypeVariable('a'), TypeVariable('b'), TypeVariable('c'))
ts := fnType.FlatTypes()
correct := Types{TypeVariable('a'), TypeVariable('b'), TypeVariable('c')}
assert.Equal(t, ts, correct)
fnType2 := NewFnType(fnType, TypeVariable('d'))
correct = append(correct, TypeVariable('d'))
ts = fnType2.FlatTypes()
assert.Equal(t, ts, correct)
}
func TestFunctionType_Clone(t *testing.T) {
fnType := NewFnType(TypeVariable('a'), TypeVariable('b'), TypeVariable('c'))
assert.Equal(t, fnType.Clone(), fnType)
rec := NewRecordType("", TypeVariable('a'), NewFnType(TypeVariable('a'), TypeVariable('b')), TypeVariable('c'))
fnType = NewFnType(rec, rec)
assert.Equal(t, fnType.Clone(), fnType)
}
|
