Python parse_cmdline Examples

Example #1

def test_parse_fetch_intraday():
	argv = ['fetch', '--intra-day', 'enag99']

	args = parse_cmdline(argv)

	assert(args.stock == "enag99")
	assert(args.start_date == datetime.date.today())
	assert(args.end_date == datetime.date.today())
	assert(args.intra_day == True)

Example #2

def test_parse_fetch_intraday():
	argv = ['fetch', 'enag99']

	args = parse_cmdline(argv)

	assert(args.stock == "enag99")
	assert(args.start_date == datetime.date.today())
	assert(args.end_date == datetime.date.today())
	assert(args.intra_day == None) # TODO: Do we want it this way?

Example #3

def test_options_fetch_intraday():
	argv = ['fetch', '--intra-day', 'enag99']

	options = get_fetch_options(parse_cmdline(argv))

	assert(options['stock'] == "ENAG99")
	assert(options['start_date'] == datetime.date.today())
	assert(options['end_date'] == datetime.date.today())
	assert(options['fetchtype'] == 'intra-day')

Example #4

def test_parse_fetch_end_date():
	argvs = [['fetch', '--end-date=3.4.2001', 'enag99'],
			 ['fetch', '-e', '3.4.2001', 'enag99']]

	for argv in argvs:
		args = parse_cmdline(argv)

		assert(args.stock == "enag99")
		assert(args.start_date == datetime.date.today())
		assert(args.end_date == datetime.date(2001, 4, 3))

Example #5

def test_parse_fetch_end_date():
	argvs = [['fetch', '--start-date=2.1.2001', '--end-date=2.3.2001', 'enag99'],
	         ['fetch', '-s', '2.1.2001', '-e', '2.3.2001', 'enag99']]
	
	for argv in argvs:
		args = parse_cmdline(argv)

		assert(args.stock == "enag99")
		assert(args.start_date == datetime.date(2001, 1, 2))
		assert(args.end_date == datetime.date(2001, 3, 2))

Example #6

def test_parse_fetch_start_date():
	argvs = [['fetch','--start-date=1.2.2001', 'enag99'],
			 ['fetch','-s', '1.2.2001', 'enag99']]

	for argv in argvs:
		args = parse_cmdline(argv)

		assert(args.stock == "enag99")
		assert(args.start_date == datetime.date(2001, 2, 1))
		assert(args.end_date == datetime.date.today())

Example #7

def test_parse_fetch_start_date_formats():
	argv = [ ['fetch', '--start-date=1.2.2001', 'enag99'],
	         ['fetch', '--start-date=2.3.2001', 'enag99'] ]
	date = [datetime.date(2001, 2, 1),
			datetime.date(2001, 3, 2)]

	assert(len(argv) == len(date))

	for idx in range(len(argv)):
		args = parse_cmdline(argv[idx])

		assert(args.stock == "enag99")
		assert(args.start_date  == date[idx])