1 USING: kernel compiler.tree.builder compiler.tree
2 compiler.tree.propagation compiler.tree.recursive
3 compiler.tree.normalization tools.test math math.order
4 accessors sequences arrays kernel.private vectors
5 alien.accessors alien.c-types sequences.private
6 byte-arrays classes.algebra classes.tuple.private
7 math.functions math.private strings layouts
8 compiler.tree.propagation.info compiler.tree.def-use
9 compiler.tree.debugger compiler.tree.checker
10 slots.private words hashtables classes assocs locals
11 specialized-arrays.double system sorting math.libm
12 math.intervals quotations effects ;
13 IN: compiler.tree.propagation.tests
15 [ V{ } ] [ [ ] final-classes ] unit-test
17 [ V{ fixnum } ] [ [ 1 ] final-classes ] unit-test
19 [ V{ fixnum } ] [ [ 1 [ ] dip ] final-classes ] unit-test
21 [ V{ fixnum object } ] [ [ 1 swap ] final-classes ] unit-test
23 [ V{ array } ] [ [ 10 f <array> ] final-classes ] unit-test
25 [ V{ array } ] [ [ { array } declare ] final-classes ] unit-test
27 [ V{ array } ] [ [ 10 f <array> swap [ ] [ ] if ] final-classes ] unit-test
29 [ V{ fixnum } ] [ [ dup fixnum? [ ] [ drop 3 ] if ] final-classes ] unit-test
31 [ V{ 69 } ] [ [ [ 69 ] [ 69 ] if ] final-literals ] unit-test
33 [ V{ fixnum } ] [ [ { fixnum } declare bitnot ] final-classes ] unit-test
35 ! Test type propagation for math ops
36 : cleanup-math-class ( obj -- class )
37 { null fixnum bignum integer ratio rational float real complex number }
38 [ class= ] with find nip ;
40 : final-math-class ( quot -- class )
41 final-classes first cleanup-math-class ;
43 [ number ] [ [ + ] final-math-class ] unit-test
45 [ bignum ] [ [ { fixnum bignum } declare + ] final-math-class ] unit-test
47 [ integer ] [ [ { fixnum integer } declare + ] final-math-class ] unit-test
49 [ bignum ] [ [ { integer bignum } declare + ] final-math-class ] unit-test
51 [ integer ] [ [ { fixnum fixnum } declare + ] final-math-class ] unit-test
53 [ float ] [ [ { float integer } declare + ] final-math-class ] unit-test
55 [ float ] [ [ { real float } declare + ] final-math-class ] unit-test
57 [ float ] [ [ { float real } declare + ] final-math-class ] unit-test
59 [ rational ] [ [ { ratio ratio } declare + ] final-math-class ] unit-test
61 [ rational ] [ [ { rational ratio } declare + ] final-math-class ] unit-test
63 [ number ] [ [ { complex complex } declare + ] final-math-class ] unit-test
65 [ float ] [ [ /f ] final-math-class ] unit-test
67 [ float ] [ [ { real real } declare /f ] final-math-class ] unit-test
69 [ integer ] [ [ /i ] final-math-class ] unit-test
71 [ integer ] [ [ { integer float } declare /i ] final-math-class ] unit-test
73 [ integer ] [ [ { float float } declare /i ] final-math-class ] unit-test
75 [ integer ] [ [ { integer } declare bitnot ] final-math-class ] unit-test
77 [ null ] [ [ { null null } declare + ] final-math-class ] unit-test
79 [ null ] [ [ { null fixnum } declare + ] final-math-class ] unit-test
81 [ float ] [ [ { float fixnum } declare + ] final-math-class ] unit-test
83 [ bignum ] [ [ { bignum bignum } declare bitxor ] final-math-class ] unit-test
85 [ float ] [ [ { float float } declare mod ] final-math-class ] unit-test
87 [ V{ fixnum } ] [ [ 255 bitand ] final-classes ] unit-test
90 [ [ 255 bitand ] [ 65535 bitand ] bi + ] final-classes
95 { fixnum } declare [ 255 bitand ] [ 65535 bitand ] bi +
100 [ { fixnum } declare [ 255 bitand ] keep + ] final-classes
104 [ { fixnum } declare 615949 * ] final-classes
108 [ 255 bitand >fixnum 3 bitor ] final-classes
112 [ >fixnum 1 mod ] final-literals
116 [ >fixnum swap [ 1 mod 69 + ] [ drop 69 ] if ] final-literals
120 [ >fixnum dup 10 > [ 1 - ] when ] final-classes
123 [ V{ integer } ] [ [ >fixnum 2 * ] final-classes ] unit-test
126 [ >fixnum dup 10 < drop 2 * ] final-classes
130 [ >fixnum dup 10 < [ 2 * ] when ] final-classes
134 [ >fixnum dup 10 < [ 2 * ] [ 2 * ] if ] final-classes
138 [ >fixnum dup 10 < [ dup -10 > [ 2 * ] when ] when ] final-classes
142 [ dup 10 < [ dup 8 > [ drop 9 ] unless ] [ drop 9 ] if ] final-literals
148 dup 10 < [ dup 8 > [ drop 9 ] unless ] [ drop 9 ] if
152 [ V{ t } ] [ [ 40 mod 40 < ] final-literals ] unit-test
154 [ V{ f } ] [ [ 40 mod 0 >= ] final-literals ] unit-test
156 [ V{ t } ] [ [ 40 rem 0 >= ] final-literals ] unit-test
158 [ V{ t } ] [ [ abs 40 mod 0 >= ] final-literals ] unit-test
160 [ t ] [ [ abs ] final-info first interval>> [0,inf] = ] unit-test
162 [ t ] [ [ absq ] final-info first interval>> [0,inf] = ] unit-test
164 [ t ] [ [ { float } declare abs ] final-info first interval>> [0,inf] = ] unit-test
166 [ t ] [ [ { float } declare absq ] final-info first interval>> [0,inf] = ] unit-test
168 [ t ] [ [ { complex } declare abs ] final-info first interval>> [0,inf] = ] unit-test
170 [ t ] [ [ { complex } declare absq ] final-info first interval>> [0,inf] = ] unit-test
172 [ t ] [ [ [ - absq ] [ + ] 2map-reduce ] final-info first interval>> [0,inf] = ] unit-test
174 [ t ] [ [ { double-array double-array } declare [ - absq ] [ + ] 2map-reduce ] final-info first interval>> [0,inf] = ] unit-test
177 [ dup string? not [ "Oops" throw ] [ ] if ] final-classes
181 [ dup string? not not >boolean [ ] [ "Oops" throw ] if ] final-classes
184 [ f ] [ [ t xor ] final-classes first null-class? ] unit-test
186 [ t ] [ [ t or ] final-classes first true-class? ] unit-test
188 [ t ] [ [ t swap or ] final-classes first true-class? ] unit-test
190 [ t ] [ [ f and ] final-classes first false-class? ] unit-test
192 [ t ] [ [ f swap and ] final-classes first false-class? ] unit-test
194 [ t ] [ [ dup not or ] final-classes first true-class? ] unit-test
196 [ t ] [ [ dup not swap or ] final-classes first true-class? ] unit-test
198 [ t ] [ [ dup not and ] final-classes first false-class? ] unit-test
200 [ t ] [ [ dup not swap and ] final-classes first false-class? ] unit-test
202 [ t ] [ [ over [ drop f ] when [ "A" throw ] unless ] final-classes first false-class? ] unit-test
207 dup [ 10 < ] [ -10 > ] bi and not [ 2 * ] unless
212 [ { fixnum } declare (clone) ] final-classes
216 [ vector new ] final-classes
221 { fixnum byte-array } declare
222 [ nth-unsafe ] 2keep [ nth-unsafe ] 2keep nth-unsafe
223 [ [ 298 * ] dip 100 * - ] dip 208 * - 128 + -8 shift
229 [ 0 dup 10 > [ 2 * ] when ] final-classes
233 [ [ 0.0 ] [ -0.0 ] if ] final-literals
237 [ /f 1.5 1.5 clamp ] final-literals
243 dup 1.5 <= [ dup 1.5 >= [ ] [ drop 1.5 ] if ] [ drop 1.5 ] if
250 dup 1.5 <= [ dup 10 >= [ ] [ drop 1.5 ] if ] [ drop 1.5 ] if
257 dup 0.0 <= [ dup 0.0 >= [ drop 0.0 ] unless ] [ drop 0.0 ] if
262 [ 0 dup 10 > [ 100 * ] when ] final-classes
266 [ 0 dup 10 > [ drop "foo" ] when ] final-classes
270 [ { fixnum } declare 3 3 - + ] final-classes
274 [ dup 10 < [ 3 * 30 < ] [ drop t ] if ] final-literals
293 [ [ "hi" ] [ 123 3 throw ] if ] final-literals
297 [ >fixnum dup 100 < [ 1 + ] [ "Oops" throw ] if ] final-classes
301 [ 0 dup 100 < not [ 1 + ] [ 1 - ] if ] final-literals
305 [ [ 1 ] [ 1 ] if 1 + ] final-literals
309 [ 0 * 10 < ] final-classes
314 123 bitand dup 10 < over 8 > and [ 3 * ] [ "B" throw ] if
320 dup number? over sequence? and [
321 dup 10 < over 8 <= not and [ 3 * ] [ "A" throw ] if
326 [ V{ string string } ] [
328 2dup [ dup string? [ "Oops" throw ] unless ] bi@ 2drop
333 [ { fixnum fixnum } declare 7 bitand neg shift ] final-classes
337 [ { fixnum } declare 1 swap 7 bitand shift ] final-classes
342 [ { fixnum } declare 1 swap 31 bitand shift ]
347 ! Array length propagation
348 [ V{ t } ] [ [ 10 f <array> length 10 = ] final-literals ] unit-test
350 [ V{ t } ] [ [ [ 10 f <array> length ] [ 10 <byte-array> length ] if 10 = ] final-literals ] unit-test
352 [ V{ t } ] [ [ [ 1 f <array> ] [ 2 f <array> ] if length 3 < ] final-literals ] unit-test
355 [ { fixnum } declare dup 10 eq? [ "A" throw ] unless ] final-literals
358 [ V{ 3 } ] [ [ [ { 1 2 3 } ] [ { 4 5 6 } ] if length ] final-literals ] unit-test
360 [ V{ 3 } ] [ [ [ B{ 1 2 3 } ] [ B{ 4 5 6 } ] if length ] final-literals ] unit-test
362 [ V{ 3 } ] [ [ [ "yay" ] [ "hah" ] if length ] final-literals ] unit-test
364 [ V{ 3 } ] [ [ 3 <byte-array> length ] final-literals ] unit-test
366 [ V{ 3 } ] [ [ 3 f <string> length ] final-literals ] unit-test
369 TUPLE: prop-test-tuple { x integer } ;
371 [ V{ integer } ] [ [ { prop-test-tuple } declare x>> ] final-classes ] unit-test
373 TUPLE: fold-boa-test-tuple { x read-only } { y read-only } { z read-only } ;
375 [ V{ T{ fold-boa-test-tuple f 1 2 3 } } ]
376 [ [ 1 2 3 fold-boa-test-tuple boa ] final-literals ]
379 TUPLE: immutable-prop-test-tuple { x sequence read-only } ;
381 [ V{ T{ immutable-prop-test-tuple f "hey" } } ] [
382 [ "hey" immutable-prop-test-tuple boa ] final-literals
386 [ { 1 2 } immutable-prop-test-tuple boa x>> ] final-literals
390 [ { array } declare immutable-prop-test-tuple boa x>> ] final-classes
394 [ complex boa ] final-classes
398 [ { float float } declare dup 0.0 <= [ "Oops" throw ] [ rect> ] if ] final-classes
401 [ V{ float float } ] [
403 { float float } declare
404 dup 0.0 <= [ "Oops" throw ] when rect>
405 [ real>> ] [ imaginary>> ] bi
411 { float float object } declare
412 [ "Oops" throw ] [ complex boa ] if
416 [ ] [ [ dup 3 slot swap 4 slot dup 3 slot swap 4 slot ] final-info drop ] unit-test
418 [ V{ number } ] [ [ [ "Oops" throw ] [ 2 + ] if ] final-classes ] unit-test
419 [ V{ number } ] [ [ [ 2 + ] [ "Oops" throw ] if ] final-classes ] unit-test
421 [ V{ POSTPONE: f } ] [
422 [ dup 1.0 <= [ drop f ] [ 0 number= ] if ] final-classes
426 TUPLE: mutable-tuple-test { x sequence } ;
429 [ "hey" mutable-tuple-test boa x>> ] final-classes
433 [ T{ mutable-tuple-test f "hey" } x>> ] final-classes
437 [ T{ mutable-tuple-test f "hey" } layout-of ] final-classes
440 ! Mixed mutable and immutable slots
441 TUPLE: mixed-mutable-immutable { x integer } { y sequence read-only } ;
443 [ V{ integer array } ] [
445 3 { 2 1 } mixed-mutable-immutable boa [ x>> ] [ y>> ] bi
449 [ V{ array integer } ] [
451 3 { 2 1 } mixed-mutable-immutable boa [ y>> ] [ x>> ] bi
455 [ V{ integer array } ] [
457 [ 2drop T{ mixed-mutable-immutable f 3 { } } ]
458 [ { array } declare mixed-mutable-immutable boa ] if
465 T{ mixed-mutable-immutable f 3 { } }
470 ! Recursive propagation
471 : recursive-test-1 ( a -- b ) recursive-test-1 ; inline recursive
473 [ V{ null } ] [ [ recursive-test-1 ] final-classes ] unit-test
475 : recursive-test-2 ( a -- b ) dup 10 < [ recursive-test-2 ] when ; inline recursive
477 [ V{ real } ] [ [ recursive-test-2 ] final-classes ] unit-test
479 : recursive-test-3 ( a -- b ) dup 10 < drop ; inline recursive
481 [ V{ real } ] [ [ recursive-test-3 ] final-classes ] unit-test
483 [ V{ real } ] [ [ [ dup 10 < ] [ ] while ] final-classes ] unit-test
486 [ { float } declare 10 [ 2.3 * ] times ] final-classes
490 [ 0 10 [ nip ] each-integer ] final-classes
494 [ t 10 [ nip 0 >= ] each-integer ] final-literals
497 : recursive-test-4 ( i n -- )
498 2dup < [ [ 1 + ] dip recursive-test-4 ] [ 2drop ] if ; inline recursive
500 [ ] [ [ recursive-test-4 ] final-info drop ] unit-test
502 : recursive-test-5 ( a -- b )
503 dup 1 <= [ drop 1 ] [ dup 1 - recursive-test-5 * ] if ; inline recursive
505 [ V{ integer } ] [ [ { integer } declare recursive-test-5 ] final-classes ] unit-test
507 : recursive-test-6 ( a -- b )
508 dup 1 <= [ drop 1 ] [ dup 1 - recursive-test-6 swap 2 - recursive-test-6 + ] if ; inline recursive
510 [ V{ integer } ] [ [ { fixnum } declare recursive-test-6 ] final-classes ] unit-test
512 : recursive-test-7 ( a -- b )
513 dup 10 < [ 1 + recursive-test-7 ] when ; inline recursive
515 [ V{ fixnum } ] [ [ 0 recursive-test-7 ] final-classes ] unit-test
517 [ V{ fixnum } ] [ [ 1 10 [ dup 10 < [ 2 * ] when ] times ] final-classes ] unit-test
519 [ V{ integer } ] [ [ 0 2 100 ^ [ nip ] each-integer ] final-classes ] unit-test
521 [ ] [ [ [ ] [ ] compose curry call ] final-info drop ] unit-test
524 [ [ drop ] [ drop ] compose curry (each-integer) ] final-classes
527 GENERIC: iterate ( obj -- next-obj ? )
528 M: fixnum iterate f ; inline
529 M: array iterate first t ; inline
531 : dead-loop ( obj -- final-obj )
532 iterate [ dead-loop ] when ; inline recursive
534 [ V{ fixnum } ] [ [ { fixnum } declare dead-loop ] final-classes ] unit-test
537 dup 0 number= [ hang-1 ] unless ; inline recursive
539 [ ] [ [ 3 hang-1 ] final-info drop ] unit-test
541 : hang-2 ( m n -- x )
550 ] if ; inline recursive
552 [ ] [ [ 3 over hang-2 ] final-info drop ] unit-test
556 dup fixnum? [ 3 over hang-2 ] [ 3 over hang-2 ] if
561 [ { hashtable } declare hashtable instance? ] final-classes
564 [ V{ POSTPONE: f } ] [
565 [ { vector } declare hashtable instance? ] final-classes
569 [ { assoc } declare hashtable instance? ] final-classes
573 [ { string } declare string? ] final-classes
576 [ V{ POSTPONE: f } ] [
577 [ 3 string? ] final-classes
581 [ { fixnum } declare [ ] curry obj>> ] final-classes
585 [ { fixnum fixnum } declare [ nth-unsafe ] curry call ] final-classes
589 [ 10 eq? [ drop 3 ] unless ] final-literals
592 GENERIC: bad-generic ( a -- b )
593 M: fixnum bad-generic 1 fixnum+fast ; inline
594 : bad-behavior ( -- b ) 4 bad-generic ; inline recursive
596 [ V{ fixnum } ] [ [ bad-behavior ] final-classes ] unit-test
600 0 10 [ bad-generic dup 123 bitand drop bad-generic 1 + ] times
604 GENERIC: infinite-loop ( a -- b )
605 M: integer infinite-loop infinite-loop ;
607 [ ] [ [ { integer } declare infinite-loop ] final-classes drop ] unit-test
609 [ V{ tuple } ] [ [ tuple-layout <tuple> ] final-classes ] unit-test
611 [ ] [ [ instance? ] final-classes drop ] unit-test
613 [ f ] [ [ V{ } clone ] final-info first literal?>> ] unit-test
615 : fold-throw-test ( a -- b ) "A" throw ; foldable
617 [ ] [ [ 0 fold-throw-test ] final-info drop ] unit-test
619 : too-deep ( a b -- c )
620 dup [ drop ] [ 2dup too-deep too-deep * ] if ; inline recursive
622 [ ] [ [ too-deep ] final-info drop ] unit-test
624 [ ] [ [ reversed boa slice boa nth-unsafe * ] final-info drop ] unit-test
628 [ ] [ [ { empty-mixin } declare empty-mixin? ] final-info drop ] unit-test
630 [ V{ fixnum } ] [ [ [ bignum-shift drop ] keep ] final-classes ] unit-test
634 [ { float float } declare complex boa ]
635 [ 2drop C{ 0.0 0.0 } ]
640 [ V{ POSTPONE: f } ] [
641 [ { float } declare 0 eq? ] final-classes
645 [ { integer fixnum } declare mod ] final-classes
649 [ { fixnum integer } declare bitand ] final-classes
652 [ V{ double-array } ] [ [| | double-array{ } ] final-classes ] unit-test
654 [ V{ t } ] [ [ netbsd unix? ] final-literals ] unit-test
656 [ V{ array } ] [ [ [ <=> ] sort [ <=> ] sort ] final-classes ] unit-test
658 [ V{ float } ] [ [ fsqrt ] final-classes ] unit-test
660 [ V{ t } ] [ [ { fixnum } declare 10 mod >float -20 > ] final-literals ] unit-test
662 [ T{ interval f { 0 t } { 127 t } } ] [
663 [ { integer } declare 127 bitand ] final-info first interval>>
667 [ [ 123 bitand ] [ drop f ] if dup [ 0 >= ] [ not ] if ] final-literals
671 [ { bignum } declare dup 1 - bitxor ] final-classes
674 [ V{ bignum integer } ] [
675 [ { bignum integer } declare [ shift ] keep ] final-classes
678 [ V{ fixnum } ] [ [ >fixnum 15 bitand 1 swap shift ] final-classes ] unit-test
680 [ V{ fixnum } ] [ [ 15 bitand 1 swap shift ] final-classes ] unit-test
683 [ { fixnum } declare log2 ] final-classes
687 [ { fixnum } declare log2 0 >= ] final-classes
690 [ V{ POSTPONE: f } ] [
691 [ { word object } declare equal? ] final-classes
695 ! [ dup string? t xor [ "A" throw ] [ ] if ] final-classes
698 ! [ t ] [ [ dup t xor or ] final-classes first true-class? ] unit-test
700 ! [ t ] [ [ dup t xor swap or ] final-classes first true-class? ] unit-test
702 ! [ t ] [ [ dup t xor and ] final-classes first false-class? ] unit-test
704 ! [ t ] [ [ dup t xor swap and ] final-classes first false-class? ] unit-test
706 ! generalize-counter-interval wasn't being called in all the right places.
707 ! bug found by littledan
709 TUPLE: littledan-1 { a read-only } ;
711 : (littledan-1-test) ( a -- ) a>> 1 + littledan-1 boa (littledan-1-test) ; inline recursive
713 : littledan-1-test ( -- ) 0 littledan-1 boa (littledan-1-test) ; inline
715 [ ] [ [ littledan-1-test ] final-classes drop ] unit-test
717 TUPLE: littledan-2 { from read-only } { to read-only } ;
719 : (littledan-2-test) ( x -- i elt )
720 [ from>> ] [ to>> ] bi + dup littledan-2 boa (littledan-2-test) ; inline recursive
722 : littledan-2-test ( x -- i elt )
723 [ 0 ] dip { array-capacity } declare littledan-2 boa (littledan-2-test) ; inline
725 [ ] [ [ littledan-2-test ] final-classes drop ] unit-test
727 : (littledan-3-test) ( x -- )
728 length 1 + f <array> (littledan-3-test) ; inline recursive
730 : littledan-3-test ( -- )
731 0 f <array> (littledan-3-test) ; inline
733 [ ] [ [ littledan-3-test ] final-classes drop ] unit-test
735 [ V{ 0 } ] [ [ { } length ] final-literals ] unit-test
737 [ V{ 1 } ] [ [ { } length 1 + f <array> length ] final-literals ] unit-test
739 ! generalize-counter is not tight enough
740 [ V{ fixnum } ] [ [ 0 10 [ 1 + >fixnum ] times ] final-classes ] unit-test
742 [ V{ fixnum } ] [ [ 0 10 [ 1 + >fixnum ] times 0 + ] final-classes ] unit-test
744 ! Coercions need to update intervals
745 [ V{ f } ] [ [ 1 2 ? 100 shift >fixnum 1 = ] final-literals ] unit-test
747 [ V{ t } ] [ [ >fixnum 1 + >fixnum most-positive-fixnum <= ] final-literals ] unit-test
749 [ V{ t } ] [ [ >fixnum 1 + >fixnum most-negative-fixnum >= ] final-literals ] unit-test
751 [ V{ f } ] [ [ >fixnum 1 + >fixnum most-negative-fixnum > ] final-literals ] unit-test
753 ! Mutable tuples with circularity should not cause problems
756 [ ] [ circle new dup >>me 1quotation final-info drop ] unit-test
758 ! Joe found an oversight
759 [ V{ integer } ] [ [ >integer ] final-classes ] unit-test
763 [ t ] [ [ foo new ] { new } inlined? ] unit-test
765 GENERIC: whatever ( x -- y )
766 M: number whatever drop foo ; inline
768 [ t ] [ [ 1 whatever new ] { new } inlined? ] unit-test
770 : that-thing ( -- class ) foo ;
772 [ f ] [ [ that-thing new ] { new } inlined? ] unit-test
774 GENERIC: whatever2 ( x -- y )
775 M: number whatever2 drop H{ { 1 1 } { 2 2 } { 3 3 } { 4 4 } { 5 6 } } ; inline
776 M: f whatever2 ; inline
778 [ t ] [ [ 1 whatever2 at ] { at* hashcode* } inlined? ] unit-test
779 [ f ] [ [ whatever2 at ] { at* hashcode* } inlined? ] unit-test
781 [ t ] [ [ { 1 2 3 } member? ] { member? } inlined? ] unit-test
782 [ f ] [ [ { 1 2 3 } swap member? ] { member? } inlined? ] unit-test
784 [ t ] [ [ { 1 2 3 } memq? ] { memq? } inlined? ] unit-test
785 [ f ] [ [ { 1 2 3 } swap memq? ] { memq? } inlined? ] unit-test
787 [ t ] [ [ V{ } clone ] { clone (clone) } inlined? ] unit-test
788 [ f ] [ [ { } clone ] { clone (clone) } inlined? ] unit-test
790 [ f ] [ [ instance? ] { instance? } inlined? ] unit-test
791 [ f ] [ [ 5 instance? ] { instance? } inlined? ] unit-test
792 [ t ] [ [ array instance? ] { instance? } inlined? ] unit-test
794 [ t ] [ [ (( a b c -- c b a )) shuffle ] { shuffle } inlined? ] unit-test
795 [ f ] [ [ { 1 2 3 } swap shuffle ] { shuffle } inlined? ] unit-test