def test_run_importanize_skip(self, mock_read): conf = { 'exclude': ['*foo.py'], } self.assertFalse( run_importanize('/path/to/importanize/file/foo.py', conf, None)) self.assertFalse(mock_read.called)
def test_run_importanize_skip(self, mock_read): conf = { 'exclude': ['*foo.py'], } self.assertFalse( run_importanize('/path/to/importanize/file/foo.py', conf, None) ) self.assertFalse(mock_read.called)
def test_run_importanize_print(self, mock_print): test_file = os.path.join(os.path.dirname(__file__), 'test_data', 'input.txt') expected_file = os.path.join(os.path.dirname(__file__), 'test_data', 'output_grouped.txt') expected = (read(expected_file) if six.PY3 else read(expected_file).encode('utf-8')) self.assertTrue( run_importanize(test_file, PEP8_CONFIG, mock.MagicMock(print=True, formatter='grouped'))) mock_print.assert_called_once_with(expected)
def test_run_importanize_write(self, mock_open): test_file = os.path.join(os.path.dirname(__file__), 'test_data', 'input.txt') expected_file = os.path.join(os.path.dirname(__file__), 'test_data', 'output_grouped.txt') expected = read(expected_file).encode('utf-8') self.assertTrue( run_importanize(test_file, PEP8_CONFIG, mock.MagicMock(print=False, formatter='grouped'))) mock_open.assert_called_once_with(test_file, 'wb') mock_open.return_value \ .__enter__.return_value \ .write.assert_called_once_with(expected)
def test_run_importanize_write_inline_group_formatter(self, mock_open): test_file = os.path.join(os.path.dirname(__file__), 'test_data', 'input.txt') expected_file = os.path.join(os.path.dirname(__file__), 'test_data', 'output_inline_grouped.txt') expected = read(expected_file).encode('utf-8') self.assertTrue( run_importanize( test_file, PEP8_CONFIG, mock.MagicMock(print=False, formatter='inline-grouped')) ) mock_open.assert_called_once_with(test_file, 'wb') mock_open.return_value \ .__enter__.return_value \ .write.assert_called_once_with(expected)
def test_run_importanize_print_inline_group_formatter(self, mock_print): test_file = os.path.join(os.path.dirname(__file__), 'test_data', 'input.txt') expected_file = os.path.join(os.path.dirname(__file__), 'test_data', 'output_inline_grouped.txt') expected = ( read(expected_file) if six.PY3 else read(expected_file).encode('utf-8') ) self.assertTrue( run_importanize( test_file, PEP8_CONFIG, mock.MagicMock(print=True, formatter='inline-grouped')) ) mock_print.assert_called_once_with(expected)