update test for python 3.10+

rodrigo-arenas · Jul 28, 2023 · be85e7d · be85e7d
1 parent 10861fd
commit be85e7d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pyworkforce/queuing/tests/test_multi_erlang.py b/pyworkforce/queuing/tests/test_multi_erlang.py
@@ -82,7 +82,8 @@ def test_expected_multierlangc_wrong_arguments():
 
     with pytest.raises(Exception) as excinfo:
         results = erlang.required_positions(arguments_grid=arguments_grid)[0]
-    assert str(excinfo.value) == "required_positions() missing 1 required positional argument: 'service_level'"
+    assert str(excinfo.value) in ["required_positions() missing 1 required positional argument: 'service_level'",
+                                  "ErlangC.required_positions() missing 1 required positional argument: 'service_level'"]
 
 
 def test_multierlangc_wrong_grid():