simLauncherTest.py 文件源码

def testFindBrainClasses(self):
        brainDir = "tmp-test-brains"
        brainModule1 = "barBrain"
        brainClass1 = "BarBrain"
        brainModule2 = "fooBrain"
        brainClass2 = "FooBrain"
        brainFile1 = os.path.join(brainDir, "%s.py" % brainModule1)
        brainFile2 = os.path.join(brainDir, "%s.py" % brainModule2)

        os.mkdir(brainDir)
        open(brainFile1, "w+").write("\n")
        open(brainFile2, "w+").write("\n")

        expected = [brainModule1 + "." + brainClass1, brainModule2 + "." + brainClass2]

        try:
            self.assertEqual(sorted(expected), sorted(simLauncher.SimulatorLauncher.findBrainClasses(brainDir)))
        finally:
            os.unlink(brainFile1)
            os.unlink(brainFile2)
            os.rmdir(brainDir)