Fix tests for float CLI input

stfc · Nov 1, 2024 · 0956286 · 0956286
1 parent 844e8c0
commit 0956286
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/tests/test_phonons_cli.py b/tests/test_phonons_cli.py
@@ -303,15 +303,15 @@ def test_supercell(supercell_arg, supercell, supercell_matrix, tmp_path):
     ("--supercell", [2]),
     ("--supercell", ["2x2x2"]),
     ("--supercell", ["2 2 2"]),
-    ("--supercell", [2.1, 2.1, 2.1]),
+    ("--supercell", ["2.1", "2.1", "2.1"]),
     ("--supercell", [2, 2, "a"]),
     ("--supercell", [2, 2]),
     ("--supercell", [2] * 6),
     ("--supercell", [2] * 9),
     ("--supercell-matrix", [2]),
     ("--supercell-matrix", ["2x2x2" * 3]),
     ("--supercell-matrix", ["2 2 2 2 2 2 2 2 2"]),
-    ("--supercell-matrix", [2.1] * 9),
+    ("--supercell-matrix", ["2.1"] * 9),
     ("--supercell-matrix", [2, 2, "a"] * 3),
     ("--supercell-matrix", [2] * 6),
     ("--supercell-matrix", [2] * 12),
@@ -337,9 +337,6 @@ def test_invalid_supercell(supercell_arg, supercell, tmp_path):
     )
     assert result.exit_code == 1 or result.exit_code == 2
 
-    print(result.exception)
-    # assert isinstance(result.exception, ValueError)
-
 
 def test_minimize_kwargs(tmp_path):
     """Test setting optimizer function and writing optimized structure."""