Add tests for parsing

tests/test_main.py

@@ -0,0 +1,13 @@
+import unittest
+
+from xena_gdc_etl import main
+
+
+class ParserTest(unittest.TestCase):
+ def setUp(self):
+ self.parser = main.create_parser()
+
+ def test_xena_eql(self):
+ parsed = self.parser.parse_args(["xena-eql", "df1", "df2"])
+ self.assertEqual(parsed.df1, "df1")
+ self.assertEqual(parsed.df2, "df2")

xena_gdc_etl/main.py

@@ -28,11 +28,11 @@ def create_parser():
  help="Test the equality of 2 Xena matrices."
  )
  equality_parser.add_argument(
- 'df1', type=str,
+ "df1", type=str,
  help='Directory for the first matrix.'
  )
  equality_parser.add_argument(
- 'df2', type=str,
+ "df2", type=str,
  help='Directory for the second matrix.'
  )
  return parser