1515# See the License for the specific language governing permissions and
1616# limitations under the License.
1717
18- import unittest2
18+ import unittest
1919import struct
2020from rsa ._compat import byte , b
2121from rsa .common import byte_size , bit_size , _bit_size
2222
2323
24- class Test_byte (unittest2 .TestCase ):
24+ class Test_byte (unittest .TestCase ):
2525 def test_values (self ):
2626 self .assertEqual (byte (0 ), b ('\x00 ' ))
2727 self .assertEqual (byte (255 ), b ('\xff ' ))
@@ -30,7 +30,7 @@ def test_struct_error_when_out_of_bounds(self):
3030 self .assertRaises (struct .error , byte , 256 )
3131 self .assertRaises (struct .error , byte , - 1 )
3232
33- class Test_byte_size (unittest2 .TestCase ):
33+ class Test_byte_size (unittest .TestCase ):
3434 def test_values (self ):
3535 self .assertEqual (byte_size (1 << 1023 ), 128 )
3636 self .assertEqual (byte_size ((1 << 1024 ) - 1 ), 128 )
@@ -55,7 +55,7 @@ def test_bad_type(self):
5555 self .assertRaises (TypeError , byte_size , "" )
5656 self .assertRaises (TypeError , byte_size , None )
5757
58- class Test_bit_size (unittest2 .TestCase ):
58+ class Test_bit_size (unittest .TestCase ):
5959 def test_zero (self ):
6060 self .assertEqual (bit_size (0 ), 0 )
6161
0 commit comments