def test_screen(self):
        from pikos.api import screen

        # default usage
        recorder = screen()
        self.assertIsInstance(recorder, TextStreamRecorder)
        self.assertIs(recorder._stream, sys.stdout)

        # with a filter
        recorder = screen(filter_=my_filter)
        self.assertIsInstance(recorder, TextStreamRecorder)
        self.assertIs(recorder._stream, sys.stdout)
        self.assertIs(recorder._filter, my_filter)
示例#2
0
文件: runner.py 项目: enthought/pikos
def main():
    description = "Execute the python script inside the pikos monitor " \
                  "context."
    parser = argparse.ArgumentParser(description=description)
    parser.add_argument('monitor', choices=MONITORS.keys(),
                        help='The monitor to use')
    parser.add_argument('-o', '--output', type=argparse.FileType('wb'),
                        help='Output results to a file')
    parser.add_argument('--buffered', action='store_false',
                        help='Use a buffered stream.')
    parser.add_argument('--recording', choices=['screen', 'text', 'csv'],
                        help='Select the type of recording to use.',
                        default='screen'),
    parser.add_argument('--focused-on', help='Provide the module path(s) of '
                        'the method where recording will be focused. '
                        'Comma separated list of importable functions',
                        default=None),
    parser.add_argument('script', help='The script to run.')
    args = parser.parse_args()

    stream = args.output if args.output is not None else sys.stdout

    if args.recording == 'text':
        recorder = textfile()
    elif args.recording == 'csv':
        if not args.buffered:
            msg = ('Unbuffered output is not supported for csv recording.'
                   'The default options for the CSVWriter will be used.')
            warnings.warn(msg)
        recorder = csvfile()
    else:
        recorder = screen()

    script = args.script
    sys.path.insert(0, os.path.dirname(script))
    focus_on = get_focused_on(script, focused_on=args.focused_on)
    monitor = MONITORS[args.monitor](recorder=recorder, focus_on=focus_on)
    run_code_under_monitor(args.script, monitor)