test coverage all examples

Yelp · Aug 30, 2016 · 1598542 · 1598542
1 parent be3568b
commit 1598542
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 5 deletions.
diff --git a/tests/examples/dbl_quote_docstring_test.py b/tests/examples/dbl_quote_docstring_test.py
@@ -39,3 +39,12 @@ def derp():
     def derp():
         """doc"""
 ''')
+
+
+def test_no_change():
+    patterns = get_patterns(dbl_quote_docstring)
+    text = """
+    def derp():
+        '''inside\"\"\"str'''
+"""
+    assert process(patterns, text) == (text)
diff --git a/undebt/examples/remove_unused_import.py b/undebt/examples/remove_unused_import.py
@@ -13,9 +13,7 @@
 @tokens_as_list(assert_len=1)
 def replace(original, location, tokens):
     header, original_without_header = tokens[0], original[len(tokens[0]):]
-    if "function" in original_without_header:
-        return None
-    elif "from function_lives_here import function\n" in header:
+    if "function" not in original_without_header and "from function_lives_here import function\n" in header:
         return header.replace("from function_lives_here import function\n", "")
-    else:
-        return None
+
+    return None