66
77from ..sexts import package_check
88
9- from nose . tools import assert_true , assert_false , assert_equal , assert_raises
9+ import pytest
1010
1111FAKE_NAME = 'nisext_improbable'
1212assert FAKE_NAME not in sys .modules
1515
1616def test_package_check ():
1717 # Try to use a required package - raise error
18- assert_raises (RuntimeError , package_check , FAKE_NAME )
18+ with pytest .raises (RuntimeError ):
19+ package_check (FAKE_NAME )
1920 # Optional, log.warn
2021 package_check (FAKE_NAME , optional = True )
2122 # Can also pass a string
@@ -29,7 +30,8 @@ def test_package_check():
2930 FAKE_MODULE .__version__ = '0.2'
3031 package_check (FAKE_NAME , version = '0.2' )
3132 # fails when version not good enough
32- assert_raises (RuntimeError , package_check , FAKE_NAME , '0.3' )
33+ with pytest .raises (RuntimeError ):
34+ package_check (FAKE_NAME , '0.3' )
3335 # Unless optional in which case log.warns
3436 package_check (FAKE_NAME , version = '0.3' , optional = True )
3537 # Might do custom version check
@@ -40,78 +42,63 @@ def test_package_check():
4042
4143def test_package_check_setuptools ():
4244 # If setuptools arg not None, missing package just adds it to arg
43- assert_raises (RuntimeError , package_check , FAKE_NAME , setuptools_args = None )
45+ with pytest .raises (RuntimeError ):
46+ package_check (FAKE_NAME , setuptools_args = None )
4447 def pkg_chk_sta (* args , ** kwargs ):
4548 st_args = {}
4649 package_check (* args , setuptools_args = st_args , ** kwargs )
4750 return st_args
48- assert_equal (pkg_chk_sta (FAKE_NAME ),
49- {'install_requires' : ['nisext_improbable' ]})
51+ assert pkg_chk_sta (FAKE_NAME ) == {'install_requires' : ['nisext_improbable' ]}
5052 # Check that this gets appended to existing value
5153 old_sta = {'install_requires' : ['something' ]}
5254 package_check (FAKE_NAME , setuptools_args = old_sta )
53- assert_equal (old_sta ,
54- {'install_requires' : ['something' , 'nisext_improbable' ]})
55+ assert old_sta == {'install_requires' : ['something' , 'nisext_improbable' ]}
5556 # That existing value as string gets converted to a list
5657 old_sta = {'install_requires' : 'something' }
5758 package_check (FAKE_NAME , setuptools_args = old_sta )
58- assert_equal (old_sta ,
59- {'install_requires' : ['something' , 'nisext_improbable' ]})
59+ assert old_sta == {'install_requires' : ['something' , 'nisext_improbable' ]}
6060 # Optional, add to extras_require
61- assert_equal (pkg_chk_sta (FAKE_NAME , optional = 'something' ),
62- {'extras_require' : {'something' : ['nisext_improbable' ]}})
61+ assert pkg_chk_sta (FAKE_NAME , optional = 'something' ) == {'extras_require' : {'something' : ['nisext_improbable' ]}}
6362 # Check that this gets appended to existing value
6463 old_sta = {'extras_require' : {'something' : ['amodule' ]}}
6564 package_check (FAKE_NAME , optional = 'something' , setuptools_args = old_sta )
66- assert_equal (old_sta ,
67- {'extras_require' :
68- {'something' : ['amodule' , 'nisext_improbable' ]}})
65+ assert old_sta == {'extras_require' : {'something' : ['amodule' , 'nisext_improbable' ]}}
6966 # That string gets converted to a list here too
7067 old_sta = {'extras_require' : {'something' : 'amodule' }}
7168 package_check (FAKE_NAME , optional = 'something' , setuptools_args = old_sta )
72- assert_equal (old_sta ,
73- {'extras_require' :
74- {'something' : ['amodule' , 'nisext_improbable' ]}})
69+ assert old_sta == {'extras_require' :
70+ {'something' : ['amodule' , 'nisext_improbable' ]}}
7571 # But optional has to be a string if not empty and setuptools_args defined
76- assert_raises (RuntimeError ,
77- package_check , FAKE_NAME , optional = True , setuptools_args = {})
72+ with pytest . raises (RuntimeError ):
73+ package_check ( FAKE_NAME , optional = True , setuptools_args = {})
7874 try :
7975 # Make a package
8076 sys .modules [FAKE_NAME ] = FAKE_MODULE
8177 # No install_requires because we already have it
82- assert_equal ( pkg_chk_sta (FAKE_NAME ), {})
78+ assert pkg_chk_sta (FAKE_NAME ) == {}
8379 # A fake version still works
8480 FAKE_MODULE .__version__ = '0.2'
85- assert_equal ( pkg_chk_sta (FAKE_NAME , version = '0.2' ), {})
81+ assert pkg_chk_sta (FAKE_NAME , version = '0.2' ) == {}
8682 # goes into install requires when version not good enough
8783 exp_spec = [FAKE_NAME + '>=0.3' ]
88- assert_equal (pkg_chk_sta (FAKE_NAME , version = '0.3' ),
89- {'install_requires' : exp_spec })
84+ assert pkg_chk_sta (FAKE_NAME , version = '0.3' ) == {'install_requires' : exp_spec }
9085 # Unless optional in which case goes into extras_require
9186 package_check (FAKE_NAME , version = '0.2' , version_getter = lambda x : '0.2' )
92- assert_equal (
93- pkg_chk_sta (FAKE_NAME , version = '0.3' , optional = 'afeature' ),
94- {'extras_require' : {'afeature' : exp_spec }})
87+ assert pkg_chk_sta (FAKE_NAME , version = '0.3' , optional = 'afeature' ) == {'extras_require' : {'afeature' : exp_spec }}
9588 # Might do custom version check
96- assert_equal (
97- pkg_chk_sta (FAKE_NAME ,
89+ assert pkg_chk_sta (FAKE_NAME ,
9890 version = '0.2' ,
99- version_getter = lambda x : '0.2' ),
100- {})
91+ version_getter = lambda x : '0.2' ) == {}
10192 # If the version check fails, put into requires
10293 bad_getter = lambda x : x .not_an_attribute
10394 exp_spec = [FAKE_NAME + '>=0.2' ]
104- assert_equal (
105- pkg_chk_sta (FAKE_NAME ,
95+ assert pkg_chk_sta (FAKE_NAME ,
10696 version = '0.2' ,
107- version_getter = bad_getter ),
108- {'install_requires' : exp_spec })
97+ version_getter = bad_getter ) == {'install_requires' : exp_spec }
10998 # Likewise for optional dependency
110- assert_equal (
111- pkg_chk_sta (FAKE_NAME ,
99+ assert pkg_chk_sta (FAKE_NAME ,
112100 version = '0.2' ,
113101 optional = 'afeature' ,
114- version_getter = bad_getter ),
115- {'extras_require' : {'afeature' : [FAKE_NAME + '>=0.2' ]}})
102+ version_getter = bad_getter ) == {'extras_require' : {'afeature' : [FAKE_NAME + '>=0.2' ]}}
116103 finally :
117104 del sys .modules [FAKE_NAME ]
0 commit comments