def _readDataList(self): """Read the only data list from 'target directory'. """ self._log(logging.INFO, "Gather input lists from %s" % self.getTargetDirectory()) #Input list dataListFiles = MyFile.dirContent(self.getTargetDirectory(), "*" + Task.OUTPUTLISTEXTENSION) #Input data or representations if len(dataListFiles) == 0: raise Exception("No data list found in %s!" % self.getTargetDirectory()) elif len(dataListFiles) > 1: raise Exception("One input list max, %d found!" % len(dataListFiles)) self._log(logging.INFO, "Found data list: %s!" % dataListFiles[0]) #Copy from target directory dataListSrcPath = self.getTargetDirectory() + os.sep + dataListFiles[0] dataListDestPath = self.getInputDirectory() + os.sep +\ MyFile.removeExtension(dataListFiles[0])[0] + Task.INPUTLISTEXTENSION MyFile.copyFile(dataListSrcPath, dataListDestPath) #Read content self.inputList = DataList() self.inputList.readFile(dataListDestPath)
def testWriteFile(self): dataList = DataList() dataList.readFile(TestDataList.IDATALIST) dataList.writeFile(TestDataList.ODATALIST) tempDataDict = copy.deepcopy(dataList.dataDictionary) dataList.readFile(TestDataList.ODATALIST) dataDict = dataList.dataDictionary self.compareDict(tempDataDict, dataDict) self.compareDict(dataDict, tempDataDict)
def testGetPath(self): dataList = DataList() dataList.readFile(TestDataList.IDATALIST) self.assertEqual('/path/to/mlf1', dataList.getPath("mlf1"))
def testReadFile(self): dataList = DataList() dataList.readFile(TestDataList.IDATALIST) self.assertEqual(15, len(dataList.dataDictionary))