minor fix for codacy

sinhaharsh · Nov 29, 2023 · fa12540 · fa12540
1 parent 4c926b1
commit fa12540
Showing 1 changed file with 9 additions and 11 deletions.
diff --git a/MRdataset/tests/test_cli.py b/MRdataset/tests/test_cli.py
@@ -32,16 +32,14 @@ def test_load_minimum_args(args):
     ds1, attributes = args
     assume(len(ds1.name) > 0)
     ds1.load()
-    with tempfile.TemporaryDirectory() as tempdir:
-        sys.argv = shlex.split(f'mrds --data-source {attributes["fake_ds_dir"]} '
-                               f'--name {ds1.name}')
-        cli()
-        output_dir = Path.cwd()
-        ds2 = load_mr_dataset(f"{output_dir}/{ds1.name}.mrds.pkl")
-        assert ds1 == ds2
-
-    with tempfile.TemporaryDirectory() as tempdir:
-        ds2 = import_dataset(data_source=attributes["fake_ds_dir"])
-        assert ds1 == ds2
+    sys.argv = shlex.split(f'mrds --data-source {attributes["fake_ds_dir"]} '
+                           f'--name {ds1.name}')
+    cli()
+    output_dir = Path.cwd()
+    ds2 = load_mr_dataset(f"{output_dir}/{ds1.name}.mrds.pkl")
+    assert ds1 == ds2
+
+    ds2 = import_dataset(data_source=attributes["fake_ds_dir"])
+    assert ds1 == ds2