cleanup tests

Jackevansevo · Apr 25, 2017 · e9211a8 · e9211a8
1 parent 07168b1
commit e9211a8
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/tests/test_primitives.py b/tests/test_primitives.py
@@ -1,7 +1,7 @@
-import pytest
+import pytest  # type: ignore
 
 from basic_utils.primitives import (
-    complement, compose, dec, even, identity, inc, odd
+    complement, comp, dec, even, identity, inc, odd
 )
 
 
@@ -13,9 +13,9 @@ def test_identity(data):
     assert identity(data) == data
 
 
-def test_compose():
-    assert compose(complement, even)(3)
-    assert compose(complement, odd)(2)
+def test_comp():
+    assert comp(complement, even)(3)
+    assert comp(complement, odd)(2)
 
 
 def test_complement():