def test_random(self): RDG = RandomDataGenerator() for i in range(10): randomData = RDG.genData() print 'randomData', i+1, ':', randomData algo1 = BruteForceSPM(randomData["T"], randomData["MS"], randomData["SDC"]) algo2 = pyMSGSP(randomData["T"], randomData["MS"], randomData["SDC"], logging.ERROR) output1 = algo1.run() output2 = algo2.run() for s in output1: for it in s: it.sort() for s in output2: for it in s: it.sort() for t in output1: self.assertTrue(t in output2)
import os import sys import inspect currentdir = os.path.dirname(os.path.abspath(inspect.getfile(inspect.currentframe()))) parentdir = os.path.dirname(currentdir) sys.path.insert(0, parentdir) from RandomDataGenerator import RandomDataGenerator DEMO_DATA_FILE = "../data/demodata.txt" DEMO_PARAM_FILE = "../data/demopara.txt" # RDG = RandomDataGenerator() RDG = RandomDataGenerator(300, 200, (0.005, 0.3), (0.005, 0.1)) data = RDG.genData() T = data["T"] MS = data["MS"] SDC = data["SDC"] with open(DEMO_DATA_FILE, "w") as df: for t in T: df.write("<") for its in t: df.write("{" + ", ".join([str(it) for it in its]) + "}") df.write(">\n") with open(DEMO_PARAM_FILE, "w") as pf: