commit 030bfac7e40b0c009d234c3d6a2ef2baf9dede5a
author Anthony Sottile <asottile@umich.edu> Thu Jan 31 19:19:10 2019 -0800
committer Anthony Sottile <asottile@umich.edu> Thu Jan 31 20:09:15 2019 -0800
tree 2330c3b6657551e498e0b0926524bf7dd4ada4ce
parent 63cc3414e9ee076267973bebb18ad2033175ce0d

Apply typing to all of pre-commit-hooks

tests/requirements_txt_fixer_test.py

diff --git a/tests/requirements_txt_fixer_test.py b/tests/requirements_txt_fixer_test.py
index 437cebd..b3a7942 100644
--- a/tests/requirements_txt_fixer_test.py
+++ b/tests/requirements_txt_fixer_test.py
@@ -1,7 +1,7 @@
 import pytest
 
 from pre_commit_hooks.requirements_txt_fixer import FAIL
-from pre_commit_hooks.requirements_txt_fixer import fix_requirements_txt
+from pre_commit_hooks.requirements_txt_fixer import main
 from pre_commit_hooks.requirements_txt_fixer import PASS
 from pre_commit_hooks.requirements_txt_fixer import Requirement
 
@@ -36,7 +36,7 @@
     path = tmpdir.join('file.txt')
     path.write_binary(input_s)
 
-    output_retval = fix_requirements_txt([path.strpath])
+    output_retval = main([path.strpath])
 
     assert path.read_binary() == output
     assert output_retval == expected_retval
@@ -44,7 +44,7 @@
 
 def test_requirement_object():
     top_of_file = Requirement()
-    top_of_file.comments.append('#foo')
+    top_of_file.comments.append(b'#foo')
     top_of_file.value = b'\n'
 
     requirement_foo = Requirement()