literal/allocations_test.rb at main · yippee-fun/literal · GitHub

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
#!/usr/bin/env ruby
# frozen_string_literal: true

# NOTE: This is not a quickdraw test becuase it needs to be run in a single thread.

require "literal"

include Literal::Types

def count_allocations
	x = GC.stat(:total_allocated_objects)
	yield
	GC.stat(:total_allocated_objects) - x
end

def count_warm_allocations(&)
	count_allocations(&)
	count_allocations(&)
	count_allocations(&)
end

def assert_allocations(type, thing)
	count_warm_allocations do
		type === thing
	end.tap do |n|
		puts "#{n} allocations for #{type.inspect} === #{thing.inspect}"
	end
end

assert_allocations(_Any, "anything") => 0
assert_allocations(_Any, false) => 0

assert_allocations(_Array(String), []) => 0
assert_allocations(_Array(String), ["a", "b"]) => 0
assert_allocations(_Array(String), ["a", "b", 1]) => 0

assert_allocations(_Boolean, true) => 0
assert_allocations(_Boolean, false) => 0
assert_allocations(_Boolean, nil) => 0
assert_allocations(_Boolean, 1) => 0

assert_allocations(_Callable, -> {}) => 0
assert_allocations(_Callable, method(:puts)) => 0

assert_allocations(_Class(Enumerable), Array) => 0

assert_allocations(_Descendant(Enumerable), Array) => 0

assert_allocations(_Enumerable(String), []) => 0

assert_allocations(_Float(18.0..), 1.234) => 0

assert_allocations(_Frozen(String), "hello") => 0

assert_allocations(_Hash(String, Integer), { "a" => 1, "b" => 2 }) => 0

assert_allocations(_Integer(18..), 18) => 0

assert_allocations(_Interface(:to_s), "Hello") => 0

assert_allocations(_Intersection(Enumerable, Array), []) => 0

assert_allocations(_JSONData, { "a" => 1, "b" => [true, false, 0, 1.23, nil, { "a" => 1 }] }) => 0

assert_allocations(_Lambda, proc {}) => 0

assert_allocations(_Map(name: String, age: Integer), { name: "Joel", age: 30 }) => 0

assert_allocations(_Never, false) => 0

assert_allocations(_Nilable(String), nil) => 0

assert_allocations(_Not(Integer), 18) => 0

assert_allocations(_Procable, proc {}) => 0

assert_allocations(_Range(Integer), 0..10) => 0

assert_allocations(_Set(String), Set["a", "b", "c"]) => 0

assert_allocations(_String(length: 5..10), "Hello") => 0

assert_allocations(_Symbol(size: 5..10), :symbol) => 0

assert_allocations(_Truthy, true) => 0
assert_allocations(_Truthy, false) => 0

assert_allocations(_Tuple(String, Integer), ["a", 1]) => 0

assert_allocations(_Union(String, "Hello", "World"), "World") => 0

assert_allocations(_Void, nil) => 0