Exemplo n.º 1
0
 def test_argparse_missing_model_name_error(self):
     with pytest.raises(TypeError, match='An arguments list is required'):
         parse_args(None)
Exemplo n.º 2
0
 def test_argparse_default_table(self):
     args = parse_args(['model_name.pkl'])
     self.assertEqual(args['table'], 'adults')
Exemplo n.º 3
0
 def test_argparse_default_valid_table_long(self):
     args = parse_args(['model_name.pkl', '--table', 'other_table'])
     self.assertEqual(args['table'], 'other_table')
Exemplo n.º 4
0
 def test_argparse_database_long(self):
     args = parse_args(
         ['model_name.pkl', '--database', 'other/database.db'])
     self.assertEqual(args['database'], 'other/database.db')
Exemplo n.º 5
0
 def test_argparse_default_database(self):
     args = parse_args(['model_name.pkl'])
     self.assertEqual(args['database'], 'lake/warehouse.db')
Exemplo n.º 6
0
 def test_argparse_model_name(self):
     args = parse_args(['model_name.pkl'])
     self.assertEqual(args['model'], 'model_name.pkl')
Exemplo n.º 7
0
 def test_argparse_short_model_name_error(self):
     with pytest.raises(ValueError):
         parse_args(['.pkl'])
Exemplo n.º 8
0
 def test_argparse_model_name_without_extension(self):
     with pytest.raises(ValueError, match='Invalid model filepath'):
         parse_args(['model_name'])