mirror of
https://github.com/vacp2p/DST-Analysis.git
synced 2026-05-10 03:00:09 -04:00
25 lines
732 B
Python
25 lines
732 B
Python
# Python Imports
|
|
import unittest
|
|
from unittest.mock import patch
|
|
|
|
# Project Imports
|
|
from src import arg_parser
|
|
|
|
|
|
class TestAnalysis(unittest.TestCase):
|
|
|
|
@patch('sys.argv', ['my_script.py', '-sp', 'test1', '-p', '41234', '-i', 'test2'])
|
|
def test_parse_args(self):
|
|
sim, port, infra = arg_parser.parse_args()
|
|
print(sim, port, infra)
|
|
self.assertEqual(sim, "test1")
|
|
self.assertEqual(port, "41234")
|
|
self.assertEqual(infra, "test2")
|
|
|
|
@patch('sys.argv', ['my_script.py'])
|
|
def test_parse_args_default(self):
|
|
sim, port, infra = arg_parser.parse_args()
|
|
self.assertEqual(sim, "/simulation_data/")
|
|
self.assertEqual(port, None)
|
|
self.assertEqual(infra, None)
|