from cpu6502 import *
import os.path
import unittest
class TestCPU6502(unittest.TestCase):
def setUp(self):
self.cpu = CPU6502()
self.cpu.pc = 0x7000
self.cpu.start = self.cpu.pc
self.debug = False
print("")
print("---------------------------")
print("test method: %s" % self._testMethodName)
def check_flags(self, neg, zero, carry):
self.assertEqual(self.cpu.flag_n, neg)
self.assertEqual(self.cpu.flag_z, zero)
self.assertEqual(self.cpu.flag_c, carry)
def check_x(self, val, neg, zero, carry):
self.assertEqual(self.cpu.reg_x, val)
self.check_flags(neg, zero, carry)
def listing(self):
self.cpu.end = self.cpu.pc
self.cpu.disass()
def run_test(self, debug=False):
self.cpu.resetcyc()
if self.debug:
self.listing()
self.cpu.run()
if self.debug:
self.cpu.registers()
print("")
print("Cycles: %d" % (self.cpu.cycles))
print("")
self.assertEqual(0xff, self.cpu.sp)
def test_irq(self):
# Interrupt aktivieren
self.cpu.flag_i = False
self.cpu.mem[0xfffe] = 0x10
self.cpu.mem[0xffff] = 0x70
# Interrupt-Ausloeser
fp = open(self.cpu.irq_file, 'w')
fp.close()
# Assembler-Programm
self.cpu.addcmd('lda #', 0x80)
# Interrupt-Routine wird ausgefuehrt
self.cpu.addcmd('ldx #', 0x10)
self.cpu.addcmd('brk')
# Interrupt-Routine
self.cpu.pc = 0x7010
self.cpu.addcmd('sta aa', 0xf0)
self.cpu.addcmd('inc aa', 0xf0)
self.cpu.addcmd('rti')
self.run_test()
self.assertEqual(0x81, self.cpu.mem[0xf0])
self.check_x(0x10, False, False, False)
self.assertFalse(os.path.exists(self.cpu.irq_file))
if __name__ == '__main__':
unittest.main()
