| 1 | #!/usr/bin/perl |
| 2 | use strict; |
| 3 | use warnings; |
| 4 | |
| 5 | use Test::More tests => 31; |
| 6 | BEGIN { use_ok('App::Scheme79asm::Compiler', qw/pretty_print/) }; |
| 7 | |
| 8 | sub is_sexp { |
| 9 | my ($expr, $expected, $name) = @_; |
| 10 | is pretty_print($expr), $expected, $name; |
| 11 | } |
| 12 | |
| 13 | sub to_sexp { |
| 14 | my ($string) = @_; |
| 15 | scalar Data::SExpression->new({fold_lists => 0, use_symbol_class => 1})->read($string) |
| 16 | } |
| 17 | |
| 18 | sub new { |
| 19 | App::Scheme79asm::Compiler->new; |
| 20 | } |
| 21 | |
| 22 | sub is_toplevel { |
| 23 | my ($string, $expected) = @_; |
| 24 | is_sexp new->process_toplevel(to_sexp $string), $expected, "process_toplevel $string"; |
| 25 | is_sexp new->compile_string($string), $expected, "compile_string $string"; |
| 26 | } |
| 27 | |
| 28 | is_sexp new->process_quoted(to_sexp '5'), '(SYMBOL 3)', 'process_quoted 5'; |
| 29 | is_sexp new->process_quoted(to_sexp '()'), '(LIST 0)', 'process_quoted ()'; |
| 30 | is_sexp new->process_quoted(to_sexp '(5 foo)'), '(LIST (LIST (LIST 0) (SYMBOL 3)) (SYMBOL 4))', 'process_quoted (5 foo)'; |
| 31 | is_sexp new->process_quoted(to_sexp '(((5)))'), '(LIST (LIST 0) (LIST (LIST 0) (LIST (LIST 0) (SYMBOL 3))))', 'process_quoted (((5)))'; |
| 32 | |
| 33 | is_toplevel '()', '(LIST 0)'; |
| 34 | is_toplevel 'NIL', '(LIST 0)'; |
| 35 | is_toplevel 'T', '(SYMBOL 2)'; |
| 36 | is_toplevel '(quote 5)', '(SYMBOL 3)'; |
| 37 | is_toplevel '(reverse-list \'a \'a \'b)', '(CALL (MORE (MORE (REVERSE-LIST 0) (SYMBOL 4)) (SYMBOL 3)) (SYMBOL 3))'; |
| 38 | is_toplevel '(if t \'(2 3) \'x)', '(IF (LIST (SYMBOL 5) (LIST (LIST (LIST 0) (SYMBOL 3)) (SYMBOL 4))) (SYMBOL 2))'; |
| 39 | is_toplevel '(car \'(1 2))', '(CALL (CAR 0) (LIST (LIST (LIST 0) (SYMBOL 3)) (SYMBOL 4)))'; |
| 40 | is_toplevel '(lambda id (x) x)', '(PROC (VAR -2))'; |
| 41 | is_toplevel '((lambda id (x) x) 5)', '(CALL (MORE (FUNCALL 0) (PROC (VAR -2))) (SYMBOL 3))'; |
| 42 | is_toplevel '(lambda append (x y) (if (atom x) y (cons (car x) (append (cdr x) y))))', '(PROC (IF (LIST (CALL (MORE (CONS 0) (CALL (MORE (MORE (FUNCALL 0) (VAR -1)) (VAR -2)) (CALL (CDR 0) (VAR -3)))) (CALL (CAR 0) (VAR -3))) (VAR -2)) (CALL (ATOM 0) (VAR -3))))'; |
| 43 | |
| 44 | sub pp_roundtrip { |
| 45 | my ($string) = @_; |
| 46 | my $pp = uc App::Scheme79asm::Compiler::pretty_print(to_sexp $string); |
| 47 | is $pp, uc($string), "pretty_print roundtrip $string"; |
| 48 | } |
| 49 | |
| 50 | pp_roundtrip '()'; |
| 51 | pp_roundtrip 't'; |
| 52 | pp_roundtrip '(lambda append (x y) (if (atom x) y (cons (car x) (append (cdr x) y))))'; |
| 53 | |
| 54 | sub expect_error_like (&$) { |
| 55 | my ($block, $error_re) = @_; |
| 56 | my $name = "test error like /$error_re/"; |
| 57 | my $result = eval { $block->(); 1 }; |
| 58 | if ($result) { |
| 59 | note 'Block did not throw an exception, failing test'; |
| 60 | fail $name; |
| 61 | } else { |
| 62 | like $@, qr/$error_re/, $name; |
| 63 | } |
| 64 | } |
| 65 | |
| 66 | expect_error_like { new->process_quoted([]) } 'argument to process_quoted is not a scalar, cons, or nil'; |
| 67 | expect_error_like { is_toplevel 'x' } 'Variable x not in environment'; |
| 68 | expect_error_like { is_toplevel '(car)' } 'Cannot call primitive car with no arguments'; |