rom test hackery

gistfile1.txt

#!/usr/bin/env python
#
# Copyright (C) 2013 Garrett Brown
# See Copyright Notice in rominfo.py

import testutils

import unittest
import os
import json

nintendo64 = testutils.loadModule("nintendo64")

class TestNintendo64Parser(unittest.TestCase):
    def setUp(self):
        self.n64Parser = nintendo64.Nintendo64Parser()

    def test_nintendo64(self):
        empty = self.n64Parser.parse("data/empty")
        self.assertEquals(len(empty), 0)

        n64_results = list()
        rom_count = 0
        failed_count = 0

        for x in os.listdir(os.path.join(os.getcwd(), "data", "N64", "ROMS")):
            print x
            path_x = os.path.join(os.getcwd(), "data", "N64", "ROMS", x)
            try:
                props = self.n64Parser.parse(path_x)
                rom_count +=1
            except Exception as e:
                props = {"failed": str(e)}
                failed_count +=1
            n64_results.append({"filename": x, "result": props})

        f = file("n64_output.json", "w")
        f.write(json.dumps({"total": rom_count, "failed": failed_count, "result": n64_results}, sort_keys=True, indent=4))
        f.close()

if __name__ == '__main__':
    unittest.main()