Mercurial > repos > yufei-luo > s_mart
comparison commons/tools/tests/Test_F_RetrieveInitHeaders.py @ 18:94ab73e8a190
Uploaded
| author | m-zytnicki |
|---|---|
| date | Mon, 29 Apr 2013 03:20:15 -0400 |
| parents | |
| children |
comparison
equal
deleted
inserted
replaced
| 17:b0e8584489e6 | 18:94ab73e8a190 |
|---|---|
| 1 from commons.core.sql.DbMySql import DbMySql | |
| 2 from commons.core.sql.TablePathAdaptator import TablePathAdaptator | |
| 3 from commons.core.coord.PathUtils import PathUtils | |
| 4 from commons.core.utils.FileUtils import FileUtils | |
| 5 import os | |
| 6 import unittest | |
| 7 | |
| 8 class Test_F_RetrieveInitHeaders(unittest.TestCase): | |
| 9 | |
| 10 def setUp(self): | |
| 11 self._iDb = DbMySql() | |
| 12 | |
| 13 def tearDown(self): | |
| 14 self._iDb.close() | |
| 15 | |
| 16 def test_run_as_script_rename_subject_and_clean_table(self): | |
| 17 shortHLinkFileName = "dummy.shortHlink" | |
| 18 self._writeShortHLinkFile(shortHLinkFileName) | |
| 19 pathTableName = "dummyInput_path" | |
| 20 self._createPathTable(pathTableName) | |
| 21 expFileName = "exp.path" | |
| 22 self._writeExpFile(expFileName) | |
| 23 obsTableName = "dummyOutput_path" | |
| 24 | |
| 25 cmd = "RetrieveInitHeaders.py -i %s -l %s -o %s -s -c -v 1" % (pathTableName, shortHLinkFileName, obsTableName) | |
| 26 os.system(cmd) | |
| 27 | |
| 28 self.assertTrue(self._iDb.doesTableExist(obsTableName)) | |
| 29 self.assertFalse(self._iDb.doesTableExist(pathTableName)) | |
| 30 obsFileName = "obs.path" | |
| 31 iTPA = TablePathAdaptator(self._iDb, obsTableName) | |
| 32 lPaths = iTPA.getListOfAllPaths() | |
| 33 PathUtils.writeListInFile(lPaths, obsFileName) | |
| 34 self.assertTrue(FileUtils.are2FilesIdentical(expFileName, obsFileName)) | |
| 35 | |
| 36 os.remove(shortHLinkFileName) | |
| 37 os.remove(expFileName) | |
| 38 os.remove(obsFileName) | |
| 39 self._iDb.dropTable(obsTableName) | |
| 40 | |
| 41 def _writeShortHLinkFile(self, fileName): | |
| 42 f = open(fileName, "w") | |
| 43 f.write("refTE_1\tPaphi_v2-B-R11932-Map16_classI-LTR-comp\t1\t5388\n") | |
| 44 f.write("refTE_2\tPaphi_v2-B-R2469-Map20_classI-LTR-comp\t1\t6385\n") | |
| 45 f.write("refTE_3\tPaphi_v2-B-R8543-Map9_classI-LTR-comp\t1\t7148\n") | |
| 46 f.write("refTE_4\tPaphi_v2-B-R33384-Map4_classI-LTR-incomp\t1\t262\n") | |
| 47 f.write("refTE_5\tPaphi_v2-B-P73.1298-Map3_classI-LTR-incomp\t1\t389\n") | |
| 48 f.write("refTE_6\tPaphi_v2-B-R36143-Map3_classI-LTR-incomp\t1\t393\n") | |
| 49 f.write("refTE_7\tPaphi_v2-B-R35261-Map5_classI-LTR-incomp\t1\t408\n") | |
| 50 f.write("refTE_8\tPaphi_v2-B-R12434-Map4_classI-LTR-incomp\t1\t420\n") | |
| 51 f.write("refTE_9\tPaphi_v2-B-R20580-Map4_classI-LTR-incomp\t1\t426\n") | |
| 52 f.write("refTE_10\tPaphi_v2-B-R5129-Map3_classI-LTR-incomp\t1\t441\n") | |
| 53 f.close() | |
| 54 | |
| 55 def _createPathTable(self, tableName): | |
| 56 fileName = "dummy.path" | |
| 57 f = open(fileName, "w") | |
| 58 f.write("7615\tchunk0030\t188432\t188611\trefTE_1\t5386\t5208\t0\t174\t97.23\n") | |
| 59 f.write("86375\tchunk0372\t155816\t157157\trefTE_1\t2213\t3575\t0\t843\t62.89\n") | |
| 60 f.write("88018\tchunk0381\t52141\t53500\trefTE_1\t2191\t3575\t0\t860\t63.3\n") | |
| 61 f.write("110377\tchunk0473\t117281\t117518\trefTE_1\t3966\t3716\t0\t230\t97.05\n") | |
| 62 f.write("39621\tchunk0169\t123563\t124003\trefTE_10\t1\t441\t0\t439\t99.55\n") | |
| 63 f.write("544710\tchunk2778\t62387\t62625\trefTE_10\t100\t327\t0\t165\t69.09\n") | |
| 64 f.write("601761\tchunk4780\t441\t559\trefTE_10\t5\t123\t0\t119\t100\n") | |
| 65 f.write("17589\tchunk0071\t58591\t59763\trefTE_2\t5139\t6384\t0\t1150\t98.12\n") | |
| 66 f.write("21050\tchunk0087\t186576\t186812\trefTE_2\t3636\t3891\t0\t232\t98.31\n") | |
| 67 f.close() | |
| 68 self._iDb.createTable(tableName, "path", fileName, True) | |
| 69 os.remove(fileName) | |
| 70 | |
| 71 def _writeExpFile(self, fileName): | |
| 72 f = open(fileName, "w") | |
| 73 f.write("7615\tchunk0030\t188432\t188611\tPaphi_v2-B-R11932-Map16_classI-LTR-comp\t5386\t5208\t0\t174\t97.230000\n") | |
| 74 f.write("86375\tchunk0372\t155816\t157157\tPaphi_v2-B-R11932-Map16_classI-LTR-comp\t2213\t3575\t0\t843\t62.890000\n") | |
| 75 f.write("88018\tchunk0381\t52141\t53500\tPaphi_v2-B-R11932-Map16_classI-LTR-comp\t2191\t3575\t0\t860\t63.300000\n") | |
| 76 f.write("110377\tchunk0473\t117281\t117518\tPaphi_v2-B-R11932-Map16_classI-LTR-comp\t3966\t3716\t0\t230\t97.050000\n") | |
| 77 f.write("39621\tchunk0169\t123563\t124003\tPaphi_v2-B-R5129-Map3_classI-LTR-incomp\t1\t441\t0\t439\t99.550000\n") | |
| 78 f.write("544710\tchunk2778\t62387\t62625\tPaphi_v2-B-R5129-Map3_classI-LTR-incomp\t100\t327\t0\t165\t69.090000\n") | |
| 79 f.write("601761\tchunk4780\t441\t559\tPaphi_v2-B-R5129-Map3_classI-LTR-incomp\t5\t123\t0\t119\t100.000000\n") | |
| 80 f.write("17589\tchunk0071\t58591\t59763\tPaphi_v2-B-R2469-Map20_classI-LTR-comp\t5139\t6384\t0\t1150\t98.120000\n") | |
| 81 f.write("21050\tchunk0087\t186576\t186812\tPaphi_v2-B-R2469-Map20_classI-LTR-comp\t3636\t3891\t0\t232\t98.310000\n") | |
| 82 f.close() | |
| 83 | |
| 84 if __name__ == "__main__": | |
| 85 unittest.main() |
