1#!/usr/bin/env python 2# 3# Copyright 2018, The Android Open Source Project 4# 5# Licensed under the Apache License, Version 2.0 (the "License"); 6# you may not use this file except in compliance with the License. 7# You may obtain a copy of the License at 8# 9# http://www.apache.org/licenses/LICENSE-2.0 10# 11# Unless required by applicable law or agreed to in writing, software 12# distributed under the License is distributed on an "AS IS" BASIS, 13# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 14# See the License for the specific language governing permissions and 15# limitations under the License. 16 17"""Unittests for atest_arg_parser.""" 18 19import unittest 20 21import atest_arg_parser 22 23 24class AtestArgParserUnittests(unittest.TestCase): 25 """Unit tests for atest_arg_parser.py""" 26 27 def test_get_args(self): 28 """Test get_args(): flatten a nested list. """ 29 parser = atest_arg_parser.AtestArgParser() 30 parser.add_argument('-t', '--test', help='Run the tests.') 31 parser.add_argument('-b', '--build', help='Run a build.') 32 parser.add_argument('--generate-baseline', help='Generate a baseline.') 33 test_args = ['-t', '--test', 34 '-b', '--build', 35 '--generate-baseline', 36 '-h', '--help'].sort() 37 self.assertEqual(test_args, parser.get_args().sort()) 38 39 40if __name__ == '__main__': 41 unittest.main() 42