# Add root folder to python paths # This must be done on every test in order to pass in Travis # import os # import sys # script_dir = os.path.dirname(os.path.abspath(__file__)) # sys.path.append(os.path.realpath(os.path.join(script_dir, '..', '..', '..'))) # # # noinspection PyPackageRequirements # from _development.helpers import DBInMemory as DB, Gamedata, Saveddata # # noinspection PyPackageRequirements # from _development.helpers_fits import RifterFit, KeepstarFit # from service.fit import Fit # # # Fake import wx # # todo: fix this # # from types import ModuleType # # wx = ModuleType("fake_module") # # sys.modules[wx.__name__] = wx # # def test_getAllFits(DB, RifterFit, KeepstarFit): # assert len(Fit.getAllFits()) == 0 # DB['db'].save(RifterFit) # assert len(Fit.getAllFits()) == 1 # DB['db'].save(KeepstarFit) # assert len(Fit.getAllFits()) == 2 # # # Cleanup after ourselves # DB['db'].remove(RifterFit) # DB['db'].remove(KeepstarFit) # # # def test_getFitsWithShip_RifterFit(DB, RifterFit): # DB['db'].save(RifterFit) # # assert Fit.getFitsWithShip(587)[0][1] == 'My Rifter Fit' # # DB['db'].remove(RifterFit)