def test_bad_id(self): employees = Employees(self.TEST_FILE) self.assertIsNotNone(employees) self.assertIsNone(employees.get_name(1))
def test_name(self): employees = Employees(self.TEST_FILE) self.assertIsNotNone(employees) self.assertEqual('frank', employees.get_name(3)) self.assertEqual('jo', employees.get_name(4))
LOGGER = logging.getLogger(__name__) if VERBOSE: LOGGER.setLevel(logging.DEBUG) # load employees from YAML E = Employees(INFILE) LOGGER.debug("infile ......................: %s", INFILE.name) LOGGER.debug("prog ........................: %s", PROG) LOGGER.debug("verbose .....................: %s", VERBOSE) LOGGER.debug("version .....................: %s", __version__) LOGGER.debug("employees ...................:") for n, t in E.employees.items(): LOGGER.debug("\t%s\t%s", n, t) T = E.get_name(3) LOGGER.debug("name for id 3 ...............: %s", T) T = E.get_by_id(3) LOGGER.debug("turnover for 3 ..............: %i", T) T = E.get_by_name('frank') LOGGER.debug("turnover for frank ..........: %i", T) T = E.get_by_year(2012) LOGGER.debug("turnover for 2012 ...........: %i", T) T = list(E.list_by_id(3)) LOGGER.debug("list turnover by id .........: %s", T) T = list(E.list_by_name('frank'))