fix one test

ikamensh · Aug 21, 2024 · a998f49 · a998f49
1 parent 651c822
commit a998f49
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/test/test_edits.py b/test/test_edits.py
@@ -556,8 +556,10 @@ def test_mixed_quote_types(state: State):
 def test_mixed_quote_types_unsafe(state: State):
     """Test that a multiline, mixed-quotes expression is transformed."""
 
-    out, count = code_editor.fstringify_code_by_line(s_in_mixed_quotes_unsafe, state)
-    assert out == s_in_mixed_quotes_unsafe
+    expected = '''f"one is \\"{'\\"'.join(one)}\\" and two is {two}"'''
+
+    out, count = process.fstringify_code_by_line(s_in_mixed_quotes_unsafe, state)
+    assert out == expected
 
 
 def test_super_call(state: State):