From 9e22faffffa20aa256a05468157404006e2e4d1e Mon Sep 17 00:00:00 2001
From: Miquel Sabaté Solà <msabate@suse.com>
Date: Thu, 21 Dec 2023 00:02:51 +0100
Subject: Initial commit
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Implemented the List scope which brings some handy macros and functions
to manipulate big lists on the NES.

Signed-off-by: Miquel Sabaté Solà <msabate@suse.com>
---
 test/suite.s | 186 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 186 insertions(+)
 create mode 100644 test/suite.s

(limited to 'test/suite.s')

diff --git a/test/suite.s b/test/suite.s
new file mode 100644
index 0000000..c302e0a
--- /dev/null
+++ b/test/suite.s
@@ -0,0 +1,186 @@
+;;;
+;; The definition of `RUN_TESTS` will be inserted automatically when running
+;; tests. Do not insert the definition manually.
+
+.include "common.s"
+
+
+main:
+.ifdef RUN_TESTS
+    jsr suite
+.endif
+halt:
+    jmp halt
+
+;;;
+;; Test suite
+
+suite:
+    jsr list_sum
+    jsr list_empty_get
+    jsr large_list
+    jsr list_set
+    jsr list_empty_set
+    jsr list_set_overflow
+
+    rts
+
+;; This is re-used in both `list_sum` and `list_set`.
+.macro LIST_SUM_AUX
+    lda #0
+    sta $90
+
+    LIST_INIT $0400
+
+    lda #2
+    jsr List::push
+    lda #4
+    jsr List::push
+    lda #1
+    jsr List::push
+
+    LIST_IT_FROM $0400
+:
+    jsr List::get
+    cpy #$FF
+    beq :+
+    clc
+    adc $90
+    sta $90
+    jmp :-
+:
+    nop
+.endmacro
+
+list_sum:
+    LIST_SUM_AUX
+@test_list_sum:
+    rts
+
+list_empty_get:
+    LIST_INIT $0400
+    jsr List::get
+    cpy #$FF
+    beq :+
+    lda #0
+    sta $90
+    jmp @test_list_empty_get
+:
+    lda #1
+    sta $90
+@test_list_empty_get:
+    rts
+
+large_list:
+    lda #0
+    sta $90
+    sta $91
+    sta $92
+
+    LIST_INIT $0400
+
+    ldx #0
+:
+    txa
+    jsr List::push
+
+    ;; Are we about to overflow the `x` register? If so, check if this was the
+    ;; first time or not. If so, then we let it overflow and loop again $FF
+    ;; times. Otherwise we will stop the loop, since we want to have a
+    ;; $200-sized list.
+    cpx #$FF
+    bne :+
+    inc $92
+    lda #2
+    cmp $92
+    beq :++
+:
+    inx
+    jmp :--
+:
+    ;; At this point we have stored this big list, let's add things up.
+    lda #0
+    sta $90
+    sta $91
+    LIST_IT_FROM $0400
+:
+    jsr List::get
+    cpy #$FF
+    beq @test_large_list
+    clc
+    adc $90
+    sta $90
+    lda #0
+    adc $91
+    sta $91
+    jmp :-
+
+@test_large_list:
+    rts
+
+list_set:
+    LIST_SUM_AUX
+
+    lda #0
+    sta $90
+
+    LIST_IT_FROM $0400
+
+    lda #3
+    jsr List::set
+    lda #5
+    jsr List::set
+    lda #2
+    jsr List::set
+
+    LIST_IT_FROM $0400
+:
+    jsr List::get
+    cpy #$FF
+    beq @test_list_set
+    clc
+    adc $90
+    sta $90
+    jmp :-
+@test_list_set:
+    rts
+
+list_empty_set:
+    LIST_INIT $0400
+    jsr List::set
+    cpy #$FF
+    beq :+
+    lda #0
+    sta $90
+    jmp @test_list_empty_set
+:
+    lda #1
+    sta $90
+@test_list_empty_set:
+    rts
+
+list_set_overflow:
+    ;; $0402 should not be set by this function, and this is to be checked; so
+    ;; zero it out before doing anything.
+    lda #0
+    ldy #0
+    sta $0402
+
+    LIST_INIT $0400
+    lda #1
+    jsr List::push
+    lda #2
+    jsr List::push
+
+    LIST_IT_FROM $0400
+    lda #2
+    jsr List::set
+    sty $90
+    lda #4
+    jsr List::set
+    sty $91
+    lda #6
+    jsr List::set
+    sty $92
+@test_list_overflow:
+    rts
-- 
cgit v1.2.3