--- a/contrib/python-zstandard/tests/test_estimate_sizes.py	Sun Apr 08 01:08:43 2018 +0200
+++ b/contrib/python-zstandard/tests/test_estimate_sizes.py	Mon Apr 09 10:13:29 2018 -0700
@@ -1,9 +1,6 @@
-try:
-    import unittest2 as unittest
-except ImportError:
-    import unittest
+import unittest
 
-import zstd
+import zstandard as zstd
 
 from . common import (
     make_cffi,
@@ -16,7 +13,3 @@
         size = zstd.estimate_decompression_context_size()
         self.assertGreater(size, 100000)
 
-    def test_compression_size(self):
-        params = zstd.get_compression_parameters(3)
-        size = zstd.estimate_compression_context_size(params)
-        self.assertGreater(size, 100000)