def test_get_module_version_list_from_file_with_fw_name(self, mock_open):
mock_open.return_value.__enter__ = mock_open
mock_open.return_value.__iter__ = mock.Mock(
return_value=iter(['mxnet']))
ref_list = [('mxnet', None, None)]
version_list = get_module_version_list_from_file('mock_file', {})
self.assertEqual(len(version_list), 1)
for i, version_dict in enumerate(version_list):
self.assertTupleEqual(ref_list[i], version_dict)
def test_get_module_version_list_from_file(self, mock_open):
mock_open.return_value.__enter__ = mock_open
mock_open.return_value.__iter__ = mock.Mock(return_value=iter([
'mxnet>=1.0.0,<=1.3.1', 'networkx>=1.11', 'numpy==1.12.0',
'defusedxml<=0.5.0', 'networkx~=1.11'
]))
ref_list = [('mxnet', '>=', '1.0.0'), ('mxnet', '<=', '1.3.1'),
('networkx', '>=', '1.11'), ('numpy', '==', '1.12.0'),
('defusedxml', '<=', '0.5.0'), ('networkx', '~=', '1.11')]
version_list = get_module_version_list_from_file('mock_file', {})
self.assertEqual(len(version_list), 6)
for i, version_dict in enumerate(version_list):
self.assertTupleEqual(ref_list[i], version_dict)
def test_get_module_version_list_from_file3(self, mock_open):
mock_open.return_value.__enter__ = mock_open
mock_open.return_value.__iter__ = mock.Mock(return_value=iter([
'# Commented line',
'tensorflow>=1.15.2,<2.0; python_version < "3.8"',
'tensorflow>=2.0; python_version >= "3.8" # Comment after line',
'numpy==1.12.0', 'defusedxml<=0.5.0', 'networkx~=1.11'
]))
ref_list = [('tensorflow', '>=', '2.0'), ('numpy', '==', '1.12.0'),
('defusedxml', '<=', '0.5.0'), ('networkx', '~=', '1.11')]
version_list = get_module_version_list_from_file(
'mock_file', {'python_version': '3.8.1'})
self.assertEqual(len(version_list), 4)
for i, version_dict in enumerate(version_list):
self.assertTupleEqual(ref_list[i], version_dict)