diff options
Diffstat (limited to 'tests/test_common.py')
| -rw-r--r-- | tests/test_common.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/tests/test_common.py b/tests/test_common.py index eba5d27..a563d21 100644 --- a/tests/test_common.py +++ b/tests/test_common.py @@ -15,13 +15,13 @@ # See the License for the specific language governing permissions and # limitations under the License. -import unittest2 +import unittest import struct from rsa._compat import byte, b from rsa.common import byte_size, bit_size, _bit_size -class Test_byte(unittest2.TestCase): +class Test_byte(unittest.TestCase): def test_values(self): self.assertEqual(byte(0), b('\x00')) self.assertEqual(byte(255), b('\xff')) @@ -30,7 +30,7 @@ class Test_byte(unittest2.TestCase): self.assertRaises(struct.error, byte, 256) self.assertRaises(struct.error, byte, -1) -class Test_byte_size(unittest2.TestCase): +class Test_byte_size(unittest.TestCase): def test_values(self): self.assertEqual(byte_size(1 << 1023), 128) self.assertEqual(byte_size((1 << 1024) - 1), 128) @@ -55,7 +55,7 @@ class Test_byte_size(unittest2.TestCase): self.assertRaises(TypeError, byte_size, "") self.assertRaises(TypeError, byte_size, None) -class Test_bit_size(unittest2.TestCase): +class Test_bit_size(unittest.TestCase): def test_zero(self): self.assertEqual(bit_size(0), 0) |