import unittest

from aiopg import _parse_version


class VersionTests(unittest.TestCase):

    def test_alpha(self):
        self.assertEqual((0, 1, 2, 'alpha', 2), _parse_version('0.1.2a2'))
        self.assertEqual((1, 2, 3, 'alpha', 0), _parse_version('1.2.3a'))

    def test_beta(self):
        self.assertEqual((0, 1, 2, 'beta', 2), _parse_version('0.1.2b2'))
        self.assertEqual((0, 1, 2, 'beta', 0), _parse_version('0.1.2b'))

    def test_rc(self):
        self.assertEqual((0, 1, 2, 'candidate', 5), _parse_version('0.1.2c5'))
        self.assertEqual((0, 1, 2, 'candidate', 0), _parse_version('0.1.2c'))

    def test_final(self):
        self.assertEqual((0, 1, 2, 'final', 0), _parse_version('0.1.2'))

    def test_invalid(self):
        self.assertRaises(ImportError, _parse_version, '0.1')
        self.assertRaises(ImportError, _parse_version, '0.1.1.2')
        self.assertRaises(ImportError, _parse_version, '0.1.1z2')

@@ -1,19 +1,17 @@
 import 
-unit
+py
 test%0A%0Afr
@@ -47,52 +47,8 @@
 n%0A%0A%0A
-class VersionTests(unittest.TestCase):%0A%0A    
 def 
@@ -62,40 +62,22 @@
 pha(
-self
 ):%0A    
-    self.assertEqual(
+assert 
 (0, 
@@ -89,25 +89,27 @@
  'alpha', 2)
-,
+ ==
  _parse_vers
@@ -126,35 +126,20 @@
 a2')
-)%0A        self.assertEqual(
+%0A    assert 
 (1, 
@@ -151,25 +151,27 @@
  'alpha', 0)
-,
+ ==
  _parse_vers
@@ -183,23 +183,19 @@
 1.2.3a')
-)
 %0A%0A
-    
+%0A
 def test
@@ -204,40 +204,22 @@
 eta(
-self
 ):%0A    
-    self.assertEqual(
+assert 
 (0, 
@@ -234,17 +234,19 @@
 eta', 2)
-,
+ ==
  _parse_
@@ -267,35 +267,20 @@
 b2')
-)
 %0A    
-    self.assertEqual(
+assert 
 (0, 
@@ -291,25 +291,27 @@
 , 'beta', 0)
-,
+ ==
  _parse_vers
@@ -323,23 +323,19 @@
 0.1.2b')
-)
 %0A%0A
-    
+%0A
 def test
@@ -342,40 +342,22 @@
 _rc(
-self
 ):%0A    
-  
-  self.assertEqual(
+assert 
 (0, 
@@ -377,17 +377,19 @@
 ate', 5)
-,
+ ==
  _parse_
@@ -410,35 +410,20 @@
 c5')
-)
 %0A    
-    self.assertEqual(
+assert 
 (0, 
@@ -439,25 +439,27 @@
 ndidate', 0)
-,
+ ==
  _parse_vers
@@ -471,23 +471,19 @@
 0.1.2c')
-)
 %0A%0A
-    
+%0A
 def test
@@ -493,40 +493,22 @@
 nal(
-self
 ):%0A    
-    self.assertEqual(
+assert 
 (0, 
@@ -524,17 +524,19 @@
 nal', 0)
-,
+ ==
  _parse_
@@ -555,15 +555,11 @@
 .2')
-)
 %0A%0A
-    
+%0A
 def 
@@ -575,35 +575,23 @@
 lid(
-self
 ):%0A    
- 
-   self.assertR
+pytest.r
 aise
@@ -632,32 +632,24 @@
 1')%0A    
-    self.assertR
+pytest.r
 aises(Im
@@ -694,24 +694,16 @@
     
-    self.assertR
+pytest.r
 aise

Convert version tests