def __init__(self, dataDir, tempDir, remote=None): """ Load input files for a scenario """ from xpedite.util.cpuInfo import decodeCpuInfo with open(os.path.join(dataDir, BASELINE_CPU_INFO_PATH)) as fileHandle: self.fullCpuInfo = json.load(fileHandle, object_hook=decodeCpuInfo) appInfoPath = os.path.join(tempDir, XPEDITE_APP_INFO_PATH) self.profileInfo = loadProfileInfo( dataDir, PROFILE_INFO_PATH, appInfoPath=appInfoPath, remote=remote )
def __init__(self, dataDir): """ Load files with expected results for comparison """ import six.moves.cPickle as pickle from xpedite.jupyter.xpediteData import XpediteDataReader with open(os.path.join(dataDir, PROBE_CMD_BASELINE_PATH), 'rb') as probeFileHandle: self.baselineProbeMap = pickle.load(probeFileHandle) # pylint: disable=c-extension-no-member with XpediteDataReader(os.path.join(dataDir, REPORT_CMD_BASELINE_PATH)) as xpediteDataReader: self.baselineProfiles = xpediteDataReader.getData(PROFILES_KEY) self.baselineProfileInfo = loadProfileInfo(dataDir, GENERATE_CMD_BASELINE_PATH)
def __init__(self, dataDir, tempDir, remote=None): """ Load input files for a scenario """ with open(os.path.join(dataDir, BASELINE_CPU_INFO_PATH)) as fileHandle: self.fullCpuInfo = json.load(fileHandle) appInfoPath = os.path.join(tempDir, XPEDITE_APP_INFO_PATH) self.profileInfo = loadProfileInfo(dataDir, PROFILE_INFO_PATH, appInfoPath=appInfoPath, remote=remote)
def test_generate_cmd_vs_baseline(scenarioName): """ Test xpedite generate by generating a new profileInfo.py file and comparing to baseline profileInfo.py in the test data directory """ from test_xpedite.test_profiler.app import TargetLauncher from xpedite.profiler.probeAdmin import ProbeAdmin with SCENARIO_LOADER[scenarioName] as scenarios: with TargetLauncher(CONTEXT, scenarios) as app: probes = ProbeAdmin.loadProbes(app.xpediteApp) profileInfoFile = generateProfileInfoFile(app.xpediteApp, probes) generatedProfileInfo = loadProfileInfo('', profileInfoFile) (generatedProfileInfo.benchmarkPaths, generatedProfileInfo.cpuSet, generatedProfileInfo.pmc) = (None, None, None) findDiff(generatedProfileInfo.__dict__, scenarios.baselineProfileInfo.__dict__) assert generatedProfileInfo == scenarios.baselineProfileInfo