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_a(self, val, neg, zero, carry):
self.assertEqual(self.cpu.reg_a, val)
self.check_flags(neg, zero, 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_brk_file(self):
# Interrupt-Ausloeser
fp = open(self.cpu.brk_file, 'w')
fp.close()
# Assembler-Programm
self.cpu.addcmd('lda #', 0x20)
# Befehle werden nicht mehr ausgefuehrt
self.cpu.addcmd('ldx #', 0x10)
self.cpu.addcmd('brk')
self.run_test()
self.check_a(0x20, False, False, False)
self.check_x(0, False, False, False)
self.assertFalse(os.path.exists(self.cpu.brk_file))
if __name__ == '__main__':
unittest.main()
