def test_setup_logging(self, rfh_mock, setLevel_mock, addHandler_mock): # Testing silent mode _, tmpoutfile = tempfile.mkstemp() main.setup_logging(False, tmpoutfile) self.assertTrue(addHandler_mock.called) setLevel_mock.assert_called_with(logging.INFO) addHandler_mock.reset_mock() setLevel_mock.reset_mock() # Testing verbose mode main.setup_logging(True, tmpoutfile) self.assertTrue(addHandler_mock.called) setLevel_mock.assert_called_with(logging.DEBUG)
import dns_service import tcp_service import wifi import main if '__main__' == __name__: main.setup_logging() dns_service.clean() tcp_service.clean() wifi.clean()
def test_seup_logging(self): main.setup_logging()
remove_unzipped(section, user_data_dir) start = 0 section -= 1 count = remaining break else: start += count misc()["opening_section"] = section misc()["opening_start"] = start z_m.sync() log.debug("Completed: section: %s; start game: %s" % (section, start)) return False total_added = 0 if __name__ == "__main__": from main import setup_logging setup_logging() db_path = os.path.join(".", "db.fs") z_m.set_db(db_path) import pentai.db.openings_book as ob_m openings_book = ob_m.OpeningsBook() while True: if build(openings_book, ".", count=100): break section = misc()["opening_section"] log.debug("built %s games so far, in section %s" % (total_added, section)) log.debug("Finished building %s games, now packing" % total_added) z_m.pack()
import unittest from main import setup_logging from preprocessing.tests import * from models.tests import * from analysis.tests import * from classifiers.tests import * from services.tests import * from api.tests import * from noise.tests import * if __name__ == "__main__": setup_logging() unittest.main()
# encoding: UTF-8 import main from dal import base_dal from test_main.constants import * def test_delete_performance_report(): base_dal.delete_performance_report(YEAR, QUARTER) if __name__ == '__main__': main.setup_logging() test_delete_performance_report()
faces.append(f) for l in objfile: tokens = l.split() t, data = tokens[0], tokens[1:] if t == "v": parse_vertex(data) elif t == "f": parse_face(data) return faces @click.command() @click.argument("objfile", type=click.File()) @click.argument("target", type=click.File(mode="wb")) @click.option("--precision", default=DEFAULT_PRECISION) @click.option("--grid-cell", default=GRID_CELL) def main(objfile, target, precision, grid_cell): faces = parse_faces(objfile) size, m = calculate_matrix(faces, precision, grid_cell) i = Image.frombytes("L", size, bytes(m)) i.save(target, format="bmp") if __name__ == "__main__": config = ConfigParser() config.read(CONFIG_FILE) setup_logging(config["Logging"]) main()
faces.append(f) for l in objfile: tokens = l.split() t, data = tokens[0], tokens[1:] if t == 'v': parse_vertex(data) elif t == 'f': parse_face(data) return faces @click.command() @click.argument('objfile', type=click.File()) @click.argument('target', type=click.File(mode='wb')) @click.option('--precision', default=DEFAULT_PRECISION) @click.option('--grid-cell', default=GRID_CELL) def main(objfile, target, precision, grid_cell): faces = parse_faces(objfile) size, m = calculate_matrix(faces, precision, grid_cell) i = Image.frombytes('L', size, bytes(m)) i.save(target, format='bmp') if __name__ == '__main__': config = ConfigParser() config.read(CONFIG_FILE) setup_logging(config['Logging']) main()
def test_setup_logging_creates_rotating_file_handler(): args = main.get_args() logger = main.setup_logging(args) assert type(logger.handlers[0]) is logging.handlers.RotatingFileHandler