diff --git a/pyrominfo/__init__.py b/pyrominfo/__init__.py index 5447983..2eb3676 100644 --- a/pyrominfo/__init__.py +++ b/pyrominfo/__init__.py @@ -1,7 +1,7 @@ # Copyright (C) 2013 Garrett Brown # See Copyright Notice in rominfo.py -from rominfo import RomInfoParser +from .rominfo import RomInfoParser __all__ = [ "RomInfo", diff --git a/pyrominfo/gameboy.py b/pyrominfo/gameboy.py index 97d5e89..037d95f 100644 --- a/pyrominfo/gameboy.py +++ b/pyrominfo/gameboy.py @@ -1,7 +1,7 @@ # Copyright (C) 2013 Garrett Brown # See Copyright Notice in rominfo.py -from rominfo import RomInfoParser +from .rominfo import RomInfoParser class GameboyParser(RomInfoParser): """ diff --git a/pyrominfo/gba.py b/pyrominfo/gba.py index fca623d..f0001ea 100644 --- a/pyrominfo/gba.py +++ b/pyrominfo/gba.py @@ -1,10 +1,10 @@ # Copyright (C) 2013 Garrett Brown # See Copyright Notice in rominfo.py -from rominfo import RomInfoParser +from .rominfo import RomInfoParser # Publishers are the same across these handhelds -from gameboy import gameboy_publishers +from .gameboy import gameboy_publishers class GBAParser(RomInfoParser): """ diff --git a/pyrominfo/genesis.py b/pyrominfo/genesis.py index 6cbd9c1..1e50cc4 100644 --- a/pyrominfo/genesis.py +++ b/pyrominfo/genesis.py @@ -1,7 +1,7 @@ # Copyright (C) 2013 Garrett Brown # See Copyright Notice in rominfo.py -from rominfo import RomInfoParser +from .rominfo import RomInfoParser class GensisParser(RomInfoParser): """ diff --git a/pyrominfo/mastersystem.py b/pyrominfo/mastersystem.py index 694fd45..ae72a70 100644 --- a/pyrominfo/mastersystem.py +++ b/pyrominfo/mastersystem.py @@ -1,7 +1,7 @@ # Copyright (C) 2013 Garrett Brown # See Copyright Notice in rominfo.py -from rominfo import RomInfoParser +from .rominfo import RomInfoParser class MasterSystemParser(RomInfoParser): """ diff --git a/pyrominfo/nes.py b/pyrominfo/nes.py index 2cf7ced..3cf7aa4 100644 --- a/pyrominfo/nes.py +++ b/pyrominfo/nes.py @@ -1,7 +1,7 @@ # Copyright (C) 2013 Garrett Brown # See Copyright Notice in rominfo.py -from rominfo import RomInfoParser +from .rominfo import RomInfoParser class NESParser(RomInfoParser): """ diff --git a/pyrominfo/nintendo64.py b/pyrominfo/nintendo64.py index 0c01162..133392c 100644 --- a/pyrominfo/nintendo64.py +++ b/pyrominfo/nintendo64.py @@ -1,7 +1,7 @@ # Copyright (C) 2013 Garrett Brown # See Copyright Notice in rominfo.py -from rominfo import RomInfoParser +from .rominfo import RomInfoParser class Nintendo64Parser(RomInfoParser): """ diff --git a/pyrominfo/snes.py b/pyrominfo/snes.py index 2d01570..7b01c81 100644 --- a/pyrominfo/snes.py +++ b/pyrominfo/snes.py @@ -14,7 +14,7 @@ # along with this program; if not, write to the Free Software # Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. -from rominfo import RomInfoParser +from .rominfo import RomInfoParser class SNESParser(RomInfoParser): """ diff --git a/tests/test_gba.py b/tests/test_gba.py index 4b0ac56..b4d320f 100644 --- a/tests/test_gba.py +++ b/tests/test_gba.py @@ -18,6 +18,7 @@ def test_gba(self): self.assertEqual(len(empty), 0) props = self.gbaParser.parse("data/Golden Sun - The Lost Age.gba") + self.assertEqual(len(props), 8) self.assertEqual(props["title"], "GOLDEN_SUN_B") self.assertEqual(props["code"], "AGFE")