1  # Python Imports
 2  import unittest
 3  from unittest.mock import patch
 4  
 5  # Project Imports
 6  from src import arg_parser
 7  
 8  
 9  class TestAnalysis(unittest.TestCase):
10  
11      @patch('sys.argv', ['my_script.py', '-sp', 'test1', '-p', '41234', '-i', 'test2'])
12      def test_parse_args(self):
13          sim, port, infra = arg_parser.parse_args()
14          print(sim, port, infra)
15          self.assertEqual(sim, "test1")
16          self.assertEqual(port, "41234")
17          self.assertEqual(infra, "test2")
18  
19      @patch('sys.argv', ['my_script.py'])
20      def test_parse_args_default(self):
21          sim, port, infra = arg_parser.parse_args()
22          self.assertEqual(sim, "/simulation_data/")
23          self.assertEqual(port, None)
24          self.assertEqual(infra, None)