ref: 9c723af8ffae03e7edbe18e6c084655fb4820f80
parent: 21b95d607705c7d88c0af2da548687c977bdf070
author: Bryan Bishop <kanzure@gmail.com>
date: Sat Mar 24 00:43:06 EDT 2012
even more crazy unit tests
--- a/extras/crystal.py
+++ b/extras/crystal.py
@@ -437,7 +437,7 @@
file_handler.close()
return rom
-def rom_interval(offset, length, strings=True):
+def rom_interval(offset, length, strings=True, debug=True):
"""returns hex values for the rom starting at offset until offset+length"""
global rom
returnable = []
@@ -473,6 +473,8 @@
"""you are too lazy to divide on your own?"""
if type(address) == str:
address = int(address, 16)
+ if 0x4000 <= address <= 0x7FFF:
+ raise Exception, "bank 1 does not exist"
return int(address) / 0x4000
def calculate_pointer(short_pointer, bank=None):
"""calculates the full address given a 4-byte pointer and bank byte"""
@@ -1145,19 +1147,19 @@
see: http://www.pokecommunity.com/showpost.php?p=4347261
"""
- if crystal: #convert to gold
+ if crystal != None: #convert to gold
if crystal <= 0x51: return crystal
if crystal == 0x52: return None
if 0x53 <= crystal <= 0x9E: return crystal-1
if crystal == 0x9F: return None
if 0xA0 <= crystal <= 0xA5: return crystal-2
- if crystal > 0xA5: raise "dunno yet if crystal has new insertions after crystal:0xA5 (gold:0xA3)"
- elif gold: #convert to crystal
+ if crystal > 0xA5: raise Exception, "dunno yet if crystal has new insertions after crystal:0xA5 (gold:0xA3)"
+ elif gold != None: #convert to crystal
if gold <= 0x51: return gold
if 0x52 <= gold <= 0x9D: return gold+1
if 0x9E <= gold <= 0xA3: return gold+2
- if gold > 0xA3: raise "dunno yet if crystal has new insertions after gold:0xA3 (crystal:0xA5)"
- else: raise "translate_command_byte needs either a crystal or gold command"
+ if gold > 0xA3: raise Exception, "dunno yet if crystal has new insertions after gold:0xA3 (crystal:0xA5)"
+ else: raise Exception, "translate_command_byte needs either a crystal or gold command"
pksv_gs = {0x00: "2call",
@@ -1469,13 +1471,13 @@
print " " + hex(address)
recursive_scripts = set([])
-def rec_parse_script_engine_script_at(address, origin=None):
+def rec_parse_script_engine_script_at(address, origin=None, debug=True):
"""this is called in parse_script_engine_script_at for recursion
when this works it should be flipped back to using the regular
parser."""
recursive_scripts.add((address, origin))
- return parse_script_engine_script_at(address, origin=origin)
-def find_broken_recursive_scripts(output=False):
+ return parse_script_engine_script_at(address, origin=origin, debug=debug)
+def find_broken_recursive_scripts(output=False, debug=True):
"""well.. these at least have a chance of maybe being broken?"""
for r in list(recursive_scripts):
script = {}@@ -3306,7 +3308,7 @@
script_parse_table[original_start_address : offset-1] = commands
return commands
-def parse_warp_bytes(some_bytes):
+def parse_warp_bytes(some_bytes, debug=True):
"""parse some number of warps from the data"""
assert len(some_bytes) % warp_byte_size == 0, "wrong number of bytes"
warps = []
@@ -3324,7 +3326,7 @@
"map_id": map_id,
})
return warps
-def parse_xy_trigger_bytes(some_bytes, bank=None, map_group=None, map_id=None):
+def parse_xy_trigger_bytes(some_bytes, bank=None, map_group=None, map_id=None, debug=True):
"""parse some number of triggers from the data"""
assert len(some_bytes) % trigger_byte_size == 0, "wrong number of bytes"
triggers = []
@@ -3354,7 +3356,7 @@
"script": script,
})
return triggers
-def parse_signpost_bytes(some_bytes, bank=None, map_group=None, map_id=None):
+def parse_signpost_bytes(some_bytes, bank=None, map_group=None, map_id=None, debug=True):
"""parse some number of signposts from the data
[Y position][X position][Function][Script pointer (2byte)]
@@ -3436,7 +3438,7 @@
spost.update(additional)
signposts.append(spost)
return signposts
-def parse_trainer_header_at(address, map_group=None, map_id=None):
+def parse_trainer_header_at(address, map_group=None, map_id=None, debug=True):
"""
[Bit no. (2byte)][Trainer group][Trainer]
[2byte pointer to Text when seen]
@@ -3493,7 +3495,7 @@
"script_talk_again": script_talk_again,
}
-def parse_people_event_bytes(some_bytes, address=None, map_group=None, map_id=None): #max of 14 people per map?
+def parse_people_event_bytes(some_bytes, address=None, map_group=None, map_id=None, debug=True): #max of 14 people per map?
"""parse some number of people-events from the data
see http://hax.iimarck.us/files/scriptingcodes_eng.htm#Scripthdr
@@ -3641,10 +3643,10 @@
standard_size = people_event_byte_size
parse_func = parse_people_event_bytes
-def parse_map_header_at(address, map_group=None, map_id=None):
+def parse_map_header_at(address, map_group=None, map_id=None, debug=True):
"""parses an arbitrary map header at some address"""
print "parsing a map header at: " + hex(address)
- bytes = rom_interval(address, map_header_byte_size, strings=False)
+ bytes = rom_interval(address, map_header_byte_size, strings=False, debug=debug)
bank = bytes[0]
tileset = bytes[1]
permission = bytes[2]
@@ -3666,13 +3668,13 @@
"fishing": fishing_group,
}
print "second map header address is: " + hex(second_map_header_address)
- map_header.update(parse_second_map_header_at(second_map_header_address))
- map_header.update(parse_map_event_header_at(map_header["event_address"], map_group=map_group, map_id=map_id))
+ map_header.update(parse_second_map_header_at(second_map_header_address, debug=debug))
+ map_header.update(parse_map_event_header_at(map_header["event_address"], map_group=map_group, map_id=map_id, debug=debug))
#maybe this next one should be under the "scripts" key?
- map_header.update(parse_map_script_header_at(map_header["script_address"], map_group=map_group, map_id=map_id))
+ map_header.update(parse_map_script_header_at(map_header["script_address"], map_group=map_group, map_id=map_id, debug=debug))
return map_header
-def parse_second_map_header_at(address, map_group=None, map_id=None):
+def parse_second_map_header_at(address, map_group=None, map_id=None, debug=True):
"""each map has a second map header"""
bytes = rom_interval(address, second_map_header_byte_size, strings=False)
border_block = bytes[0]
@@ -3704,7 +3706,7 @@
"connections": connections,
}
-def parse_map_event_header_at(address, map_group=None, map_id=None):
+def parse_map_event_header_at(address, map_group=None, map_id=None, debug=True):
"""parse crystal map event header byte structure thing"""
returnable = {}@@ -3745,7 +3747,7 @@
return returnable
-def parse_map_script_header_at(address, map_group=None, map_id=None):
+def parse_map_script_header_at(address, map_group=None, map_id=None, debug=True):
"""parses a script header
This structure allows the game to have e.g. one-time only events on a map
@@ -3872,7 +3874,7 @@
map_header_offset = offset + ((map_id - 1) * map_header_byte_size)
return parse_map_header_at(map_header_offset, map_group=map_group, map_id=map_id)
-def parse_all_map_headers():
+def parse_all_map_headers(debug=True):
"""calls parse_map_header_at for each map in each map group"""
global map_names
if not map_names[1].has_key("offset"):@@ -3883,9 +3885,9 @@
#del group_data["offset"]
for map_id, map_data in group_data.items():
if map_id == "offset": continue #skip the "offset" address for this map group
- print "map_group is: " + str(group_id) + " map_id is: " + str(map_id)
+ if debug: print "map_group is: " + str(group_id) + " map_id is: " + str(map_id)
map_header_offset = offset + ((map_id - 1) * map_header_byte_size)
- parsed_map = parse_map_header_at(map_header_offset, map_group=group_id, map_id=map_id)
+ parsed_map = parse_map_header_at(map_header_offset, map_group=group_id, map_id=map_id, debug=debug)
map_names[group_id][map_id].update(parsed_map)
map_names[group_id][map_id]["header_offset"] = map_header_offset
@@ -4433,6 +4435,9 @@
correct = "9f2922b235a5eeb78d65594e82ef5dde"
md5sum = md5.md5(rom).hexdigest()
self.assertEqual(md5sum, correct)
+ def test_bizarre_http_presence(self):
+ rom_segment = self.rom[0x112116:0x112116+8]
+ self.assertEqual(rom_segment, "HTTP/1.0")
def test_rom_interval(self):
address = 0x100
interval = 10
@@ -4455,6 +4460,47 @@
def test_how_many_until(self):
how_many = how_many_until(chr(0x13), 0x1337)
self.assertEqual(how_many, 3)
+ def test_calculate_bank(self):
+ self.failUnless(calculate_bank(0x8000) == 2)
+ self.failUnless(calculate_bank("0x9000") == 2)+ self.failUnless(calculate_bank(0) == 0)
+ for address in [0x4000, 0x5000, 0x6000, 0x7000]:
+ self.assertRaises(Exception, calculate_bank, address)
+ def test_calculate_pointer(self):
+ pass #or raise NotImplementedError, bryan_message
+ def test_calculate_pointer_from_bytes_at(self):
+ pass #or raise NotImplementedError, bryan_message
+ def test_rom_text_at(self):
+ self.assertEquals(rom_text_at(0x112116, 8), "HTTP/1.0")
+ def test_find_all_text_pointers_in_script_engine_script(self):
+ "finds text pointers from scripts"
+ pass #or raise NotImplementedError, bryan_message
+ def test_translate_command_byte(self):
+ self.failUnless(translate_command_byte(crystal=0x0) == 0x0)
+ self.failUnless(translate_command_byte(crystal=0x10) == 0x10)
+ self.failUnless(translate_command_byte(crystal=0x40) == 0x40)
+ self.failUnless(translate_command_byte(gold=0x0) == 0x0)
+ self.failUnless(translate_command_byte(gold=0x10) == 0x10)
+ self.failUnless(translate_command_byte(gold=0x40) == 0x40)
+ self.assertEqual(translate_command_byte(gold=0x0), translate_command_byte(crystal=0x0))
+ self.failUnless(translate_command_byte(gold=0x52) == 0x53)
+ self.failUnless(translate_command_byte(gold=0x53) == 0x54)
+ self.failUnless(translate_command_byte(crystal=0x53) == 0x52)
+ self.failUnless(translate_command_byte(crystal=0x52) == None)
+ self.assertRaises(Exception, translate_command_byte, None, gold=0xA4)
+ def test_pksv_integrity(self):
+ "does pksv_gs look okay?"
+ self.assertEqual(pksv_gs[0x00], "2call")
+ self.assertEqual(pksv_gs[0x2D], "givepoke")
+ self.assertEqual(pksv_gs[0x85], "waitbutton")
+ self.assertEqual(pksv_crystal[0x00], "2call")
+ self.assertEqual(pksv_crystal[0x86], "waitbutton")
+ self.assertEqual(pksv_crystal[0xA2], "credits")
+ def test_chars_integrity(self):
+ self.assertEqual(chars[0x80], "A")
+ self.assertEqual(chars[0xA0], "a")
+ self.assertEqual(chars[0xF0], "¥")
+ self.assertEqual(jap_chars[0x44], "ぱ")
class TestRomStr(unittest.TestCase):
"""RomStr is a class that should act exactly like str()
except that it never shows the contents of it string
@@ -4476,6 +4522,64 @@
def test_conversion(self):
"check if RomStr() -> str() works"
self.assertEquals(str(self.sample), self.sample_text)
+class TestMapParsing(unittest.TestCase):
+ def test_parse_warp_bytes(self):
+ pass #or raise NotImplementedError, bryan_message
+ def test_parse_xy_trigger_bytes(self):
+ pass #or raise NotImplementedError, bryan_message
+ def test_parse_signpost_bytes(self):
+ pass #or raise NotImplementedError, bryan_message
+ def test_parse_people_event_bytes(self):
+ pass #or raise NotImplementedError, bryan_message
+ def test_parse_trainer_header_at(self):
+ pass #or raise NotImplementedError, bryan_message
+ def test_parse_map_header_at(self):
+ pass #or raise NotImplementedError, bryan_message
+ def test_parse_second_map_header_at(self):
+ pass #or raise NotImplementedError, bryan_message
+ def test_parse_map_event_header_at(self):
+ pass #or raise NotImplementedError, bryan_message
+ def test_parse_map_script_header_at(self):
+ pass #or raise NotImplementedError, bryan_message
+ def test_parse_map_header_by_id(self):
+ pass #or raise NotImplementedError, bryan_message
+ def test_parse_all_map_headers(self):
+ global parse_map_header_at, counter
+ counter = 0
+ for k in map_names.keys():
+ if "offset" not in map_names[k].keys():
+ map_names[k]["offset"] = 0
+ temp = parse_map_header_at
+ def parse_map_header_at(address, map_group=None, map_id=None, debug=False):
+ global counter
+ counter += 1
+ return {}+ parse_all_map_headers(debug=False)
+ self.assertEqual(counter, 388)
+ parse_map_header_at = temp
+class TestTextScript(unittest.TestCase):
+ def test_find_addresses(self):
+ pass #or raise NotImplementedError, bryan_message
+ def test_parse_text_at(self):
+ pass #or raise NotImplementedError, bryan_message
+ def test_to_asm_at(self):
+ pass #or raise NotImplementedError, bryan_message
+ def test_to_asm(self):
+ pass #or raise NotImplementedError, bryan_message
+class TestEncodedText(unittest.TestCase):
+ def test_to_asm(self):
+ pass #or raise NotImplementedError, bryan_message
+ def test_process_00_subcommands(self):
+ pass #or raise NotImplementedError, bryan_message
+ def test_from_bytes(self):
+ pass #or raise NotImplementedError, bryan_message
+ def test_parse_text_at(self):
+ pass #or raise NotImplementedError, bryan_message
+class TestScript(unittest.TestCase):
+ """for testing parse_script_engine_script_at
+ and script parsing in general.
+ Script should be a class?"""
+ pass
class TestMetaTesting(unittest.TestCase):
"""test whether or not i am finding at least
some of the tests in this file"""
--
⑨