123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100 |
- import argparse
- import pytest
- import deepspeed
- def basic_parser():
- parser = argparse.ArgumentParser()
- parser.add_argument('--num_epochs', type=int)
- return parser
- def test_no_ds_arguments_no_ds_parser():
- parser = basic_parser()
- args = parser.parse_args(['--num_epochs', '2'])
- assert args
- assert hasattr(args, 'num_epochs')
- assert args.num_epochs == 2
- assert not hasattr(args, 'deepspeed')
- assert not hasattr(args, 'deepspeed_config')
- def test_no_ds_arguments():
- parser = basic_parser()
- parser = deepspeed.add_config_arguments(parser)
- args = parser.parse_args(['--num_epochs', '2'])
- assert args
- assert hasattr(args, 'num_epochs')
- assert args.num_epochs == 2
- assert hasattr(args, 'deepspeed')
- assert args.deepspeed == False
- assert hasattr(args, 'deepspeed_config')
- assert args.deepspeed_config == None
- def test_no_ds_enable_argument():
- parser = basic_parser()
- parser = deepspeed.add_config_arguments(parser)
- args = parser.parse_args(['--num_epochs', '2', '--deepspeed_config', 'foo.json'])
- assert args
- assert hasattr(args, 'num_epochs')
- assert args.num_epochs == 2
- assert hasattr(args, 'deepspeed')
- assert args.deepspeed == False
- assert hasattr(args, 'deepspeed_config')
- assert type(args.deepspeed_config) == str
- assert args.deepspeed_config == 'foo.json'
- def test_no_ds_config_argument():
- parser = basic_parser()
- parser = deepspeed.add_config_arguments(parser)
- args = parser.parse_args(['--num_epochs', '2', '--deepspeed'])
- assert args
- assert hasattr(args, 'num_epochs')
- assert args.num_epochs == 2
- assert hasattr(args, 'deepspeed')
- assert type(args.deepspeed) == bool
- assert args.deepspeed == True
- assert hasattr(args, 'deepspeed_config')
- assert args.deepspeed_config == None
- def test_no_ds_parser():
- parser = basic_parser()
- with pytest.raises(SystemExit):
- args = parser.parse_args(['--num_epochs', '2', '--deepspeed'])
- def test_core_deepscale_arguments():
- parser = basic_parser()
- parser = deepspeed.add_config_arguments(parser)
- args = parser.parse_args(
- ['--num_epochs',
- '2',
- '--deepspeed',
- '--deepspeed_config',
- 'foo.json'])
- assert args
- assert hasattr(args, 'num_epochs')
- assert args.num_epochs == 2
- assert hasattr(args, 'deepspeed')
- assert type(args.deepspeed) == bool
- assert args.deepspeed == True
- assert hasattr(args, 'deepspeed_config')
- assert type(args.deepspeed_config) == str
- assert args.deepspeed_config == 'foo.json'
|