Python load_valuesの例

コード例 #1

ファイル: test_scripts.py プロジェクト: kaman1/college-costs

 def test_load_values(self, mock_process_missing, mock_process):
     mock_process.return_value = {'999999': {'onCampusAvail': '2'}}
     msg = update_ipeds.load_values()
     self.assertTrue('DRY' in msg)
     self.assertTrue(mock_process.call_count == 1)
     mock_process.return_value = {
         '243197': {
             'onCampusAvail': '2',
             'books': '.'
         }
     }
     msg = update_ipeds.load_values()
     self.assertTrue('DRY' in msg)
     self.assertTrue(mock_process.call_count == 2)
     msg = update_ipeds.load_values(dry_run=False)
     self.assertFalse('DRY' in msg)
     self.assertTrue(mock_process.call_count == 3)
     mock_process.return_value = {'243197': {'onCampusAvail': '1'}}
     msg = update_ipeds.load_values()
     self.assertTrue('DRY' in msg)
     self.assertTrue(mock_process.call_count == 4)
     mock_process.return_value = {'999998': {'onCampusAvail': '2'}}
     msg = update_ipeds.load_values(dry_run=False)
     self.assertFalse('DRY' in msg)
     self.assertTrue(mock_process.call_count == 5)
     self.assertTrue(mock_process_missing.call_count == 1)

コード例 #2

ファイル: update_ipeds.py プロジェクト: cfpb/college-costs

 def handle(self, *args, **options):
     if options["dry_run"].lower() == "false":
         msg = load_values(dry_run=False)
     elif options["dry_run"].lower() == "true":
         msg = load_values()
     else:
         msg = PARSER_HELP
     self.stdout.write(msg)

コード例 #3

ファイル: update_ipeds.py プロジェクト: ultrageek/college-costs

 def handle(self, *args, **options):
     if options['dry_run'].lower() == 'false':
         msg = load_values(dry_run=False)
     elif options['dry_run'].lower() == 'true':
         msg = load_values()
     else:
         msg = PARSER_HELP
     self.stdout.write(msg)

コード例 #4

ファイル: test_scripts.py プロジェクト: cfpb/college-costs

 def test_load_values(self, mock_process_missing, mock_process):
     mock_process.return_value = {'999999': {'onCampusAvail': '2'}}
     msg = update_ipeds.load_values()
     self.assertTrue('DRY' in msg)
     self.assertTrue(mock_process.call_count == 1)
     mock_process.return_value = {'243197': {'onCampusAvail': '2',
                                             'books': '.'}}
     msg = update_ipeds.load_values()
     self.assertTrue('DRY' in msg)
     self.assertTrue(mock_process.call_count == 2)
     msg = update_ipeds.load_values(dry_run=False)
     self.assertFalse('DRY' in msg)
     self.assertTrue(mock_process.call_count == 3)
     mock_process.return_value = {'243197': {'onCampusAvail': '1'}}
     msg = update_ipeds.load_values()
     self.assertTrue('DRY' in msg)
     self.assertTrue(mock_process.call_count == 4)
     mock_process.return_value = {'999998': {'onCampusAvail': '2'}}
     msg = update_ipeds.load_values(dry_run=False)
     self.assertFalse('DRY' in msg)
     self.assertTrue(mock_process.call_count == 5)
     self.assertTrue(mock_process_missing.call_count == 1)