Best Python code snippet using autotest_python
test_mongo_data_set_io_reader.py
Source: test_mongo_data_set_io_reader.py
1import unittest2from threading import ThreadError3from unittest.mock import patch, MagicMock4import numpy as np5from pymongo.errors import OperationFailure6from qilib.data_set import MongoDataSetIOReader, DataSet, DataArray, MongoDataSetIO7class TestMongoDataSetIOReader(unittest.TestCase):8 def test_sync_from_storage_meta_data(self):9 mock_queue = MagicMock()10 with patch('qilib.data_set.mongo_data_set_io_reader.MongoDataSetIO') as mock_io, patch(11 'qilib.data_set.mongo_data_set_io_reader.Thread') as thread, \12 patch('qilib.data_set.mongo_data_set_io_reader.Queue', return_value=mock_queue):13 reader = MongoDataSetIOReader(name='test')14 thread.assert_called_once()15 mock_io.assert_called_once_with('test', None, create_if_not_found=False, collection='data_sets',16 database='qilib')17 data_set = DataSet(storage_reader=reader)18 mock_queue.get.return_value = {'updateDescription': {'updatedFields': {'metadata': {'name': 'test_name'}}}}19 data_set.sync_from_storage(-1)20 self.assertEqual('test_name', data_set.name)21 mock_queue.get.return_value = {22 'updateDescription': {'updatedFields': {'metadata.default_array_name': 'some_array'}}}23 data_set.sync_from_storage(-1)24 self.assertEqual('some_array', data_set.default_array_name)25 def test_sync_from_storage_array_update(self):26 mock_queue = MagicMock()27 with patch('qilib.data_set.mongo_data_set_io_reader.MongoDataSetIO') as mock_io, patch(28 'qilib.data_set.mongo_data_set_io_reader.Thread') as thread, \29 patch('qilib.data_set.mongo_data_set_io_reader.Queue', return_value=mock_queue):30 reader = MongoDataSetIOReader(name='test')31 thread.assert_called_once()32 mock_io.assert_called_once_with('test', None, create_if_not_found=False, collection='data_sets',33 database='qilib')34 data_set = DataSet(storage_reader=reader)35 data_array = DataArray(name='test_array', label='lab', shape=(2, 2))36 data_set.add_array(data_array)37 mock_queue.get.return_value = {'updateDescription': {38 'updatedFields': {'array_updates': [[[0, 0], {'test_array': 42}], [[0, 1], {'test_array': 25}]]}}}39 data_set.sync_from_storage(-1)40 self.assertListEqual([42, 25], list(data_set.test_array[0]))41 mock_queue.get.return_value = {'updateDescription': {42 'updatedFields': {'array_updates.1': [1, {'test_array': [67, 67]}]}}}43 data_set.sync_from_storage(-1)44 self.assertListEqual([67, 67], list(data_set.test_array[1]))45 def test_sync_from_storage_array(self):46 mock_queue = MagicMock()47 with patch('qilib.data_set.mongo_data_set_io_reader.MongoDataSetIO') as mock_io, patch(48 'qilib.data_set.mongo_data_set_io_reader.Thread') as thread, \49 patch('qilib.data_set.mongo_data_set_io_reader.Queue', return_value=mock_queue):50 mock_io.decode_numpy_array = MongoDataSetIO.decode_numpy_array51 reader = MongoDataSetIOReader(name='test')52 thread.assert_called_once()53 mock_io.assert_called_once_with('test', None, create_if_not_found=False, collection='data_sets',54 database='qilib')55 data_set = DataSet(storage_reader=reader)56 set_array = DataArray(name='setter', label='for_testing', is_setpoint=True,57 preset_data=np.array(range(0, 2)))58 set_array[0] = 4259 set_array[1] = 2560 data_array = DataArray(name='test_array', label='lab', unit='V', is_setpoint=False, set_arrays=[set_array],61 shape=(2,))62 update_data = {"data_arrays": {"setter": {63 "name": set_array.name,64 "label": set_array.label,65 "unit": set_array.unit,66 "is_setpoint": set_array.is_setpoint,67 "set_arrays": [array.name for array in set_array.set_arrays],68 "preset_data": MongoDataSetIO.encode_numpy_array(set_array)}}}69 mock_queue.get.return_value = {'updateDescription': {'updatedFields': update_data}}70 data_set.sync_from_storage(-1)71 update_data = {"data_arrays.test_array": {72 "name": data_array.name,73 "label": data_array.label,74 "unit": data_array.unit,75 "is_setpoint": data_array.is_setpoint,76 "set_arrays": [array.name for array in data_array.set_arrays],77 "preset_data": MongoDataSetIO.encode_numpy_array(data_array)}}78 mock_queue.get.return_value = {'updateDescription': {'updatedFields': update_data}}79 data_set.sync_from_storage(-1)80 self.assertEqual('test_array', data_set.test_array.name)81 self.assertEqual('lab', data_set.test_array.label)82 self.assertEqual('V', data_set.test_array.unit)83 self.assertFalse(data_set.test_array.is_setpoint)84 self.assertEqual('setter', data_array.set_arrays[0].name)85 self.assertListEqual([42, 25], list(data_array.set_arrays[0]))86 data_array[0] = 25587 update_data["data_arrays.test_array"]["preset_data"] = MongoDataSetIO.encode_numpy_array(data_array)88 data_set.sync_from_storage(-1)89 self.assertEqual(255, data_set.test_array[0])90 def test_sync_from_storage_error_on_queue(self):91 mock_queue = MagicMock()92 with patch('qilib.data_set.mongo_data_set_io_reader.MongoDataSetIO') as mock_io, patch(93 'qilib.data_set.mongo_data_set_io_reader.Thread') as thread, \94 patch('qilib.data_set.mongo_data_set_io_reader.Queue', return_value=mock_queue):95 reader = MongoDataSetIOReader(name='test')96 thread.assert_called_once()97 mock_io.assert_called_once_with('test', None, create_if_not_found=False, collection='data_sets',98 database='qilib')99 mock_queue.get.return_value = {'thread_error': OperationFailure('Error')}100 error = ThreadError, 'Watcher thread has stopped unexpectedly.'101 self.assertRaisesRegex(*error, reader.sync_from_storage, -1)102 def test_sync_from_storage_array_update_timeout(self):103 with patch('qilib.data_set.mongo_data_set_io_reader.MongoDataSetIO') as mock_io, patch(104 'qilib.data_set.mongo_data_set_io_reader.Thread') as thread:105 reader = MongoDataSetIOReader(name='test')106 thread.assert_called_once()107 mock_io.assert_called_once_with('test', None, create_if_not_found=False, collection='data_sets',108 database='qilib')109 error = TimeoutError, ''110 self.assertRaisesRegex(*error, reader.sync_from_storage, 0.001)111 def test_bind_data_set(self):112 mock_mongo_data_set_io = MagicMock()113 with patch('qilib.data_set.mongo_data_set_io_reader.MongoDataSetIO',114 return_value=mock_mongo_data_set_io) as mock_io, \115 patch('qilib.data_set.mongo_data_set_io_reader.Thread') as thread:116 mock_mongo_data_set_io.get_document.return_value = {'name': 'test',117 'metadata': {'default_array_name': 'array'}}118 reader = MongoDataSetIOReader(name='test')119 thread.assert_called_once()120 mock_io.assert_called_once_with('test', None, create_if_not_found=False, collection='data_sets',121 database='qilib')122 data_set = DataSet()123 reader.bind_data_set(data_set)124 self.assertEqual('test', data_set.name)125 self.assertEqual('array', data_set.default_array_name)126 def test_load(self):127 with patch('qilib.data_set.mongo_data_set_io_reader.MongoDataSetIO') as mock_io, \128 patch('qilib.data_set.mongo_data_set_io_reader.Thread') as thread:129 data_set = MongoDataSetIOReader.load('test', '0x2A')130 thread.assert_called_once()131 mock_io.assert_called_once_with('test', '0x2A', create_if_not_found=False, collection='data_sets',132 database='qilib')133 self.assertIsInstance(data_set.storage, MongoDataSetIOReader)134 def test_thread_and_queue(self):135 mock_queue_instance = MagicMock()136 with patch('qilib.data_set.mongo_data_set_io_reader.MongoDataSetIO'), patch(137 'qilib.data_set.mongo_data_set_io_reader.Thread') as mock_thread, \138 patch('qilib.data_set.mongo_data_set_io_reader.Queue', return_value=mock_queue_instance):139 def work(target, args):140 target(*args)141 return mock_thread142 mock_thread.side_effect = work143 iteration_error = StopIteration('Boom!')144 mock_queue_instance.put.side_effect = [iteration_error, None]145 MongoDataSetIOReader(name='test')...
test_main.py
Source: test_main.py
...26 patch.start()27 def tearDown(self):28 for patch in self.patchers:29 patch.stop()30 def mock_io(self, body: str="", status: int=0):31 mock_io = mock.MagicMock()32 mock_io.read = lambda: body.encode("utf8")33 mock_io.channel.recv_exit_status.return_value = 034 return mock_io35class TestMain(BaseTestMain):36 argv = "lfsccm"37 def test_main_no_action(self):38 self.assertEqual(-1, main.main())39@mock.patch("paramiko.client.SSHClient.connect")40@mock.patch("paramiko.client.SSHClient.exec_command")41class TestMainCheckFiles(BaseTestMain):42 # FIXME: use something like optparse to add options in anywhare43 # in the args, and handle syntax errors to notify what was wrong44 # with in the errors.45 argv = "lfsccm --files=/sample/file:ro:0 check-files"46 @mock.patch("sys.argv", ["lfsccm", "check-files"])47 def test_check_files_no_files_specified(self, mock_exec, mock_connect):48 # FIXME: exit 1 because no requremet args found49 self.assertEqual(0, main.main())50 def test_check_files_with_args(self, mock_exec, mock_connect):51 mock_exec.return_value = [52 None, self.mock_io(), self.mock_io()]53 self.assertEqual(0, main.main())54 def test_check_files_not_exist(self, mock_exec, mock_connect):55 failed_io = self.mock_io("No such file or directory")56 failed_io.channel.recv_exit_status.return_value = 157 mock_exec.return_value = [58 None, failed_io, self.mock_io()]59 self.assertEqual(1, main.main())60@mock.patch("paramiko.client.SSHClient.connect")61@mock.patch("paramiko.client.SSHClient.exec_command")62class TestMainROAvailable(BaseTestMain):63 argv = "lfsccm check-ro-available"64 def test_check_ro_available(self, mock_exec, mock_connect):65 mock_exec.return_value = [66 None, self.mock_io("Lustre version: 2.16.0"), self.mock_io()]67 self.assertEqual(0, main.main())68 def test_check_ro_available_not_supported(69 self, mock_exec, mock_connect):70 # lustre 2.14 doesn't support RO mode of PCC71 mock_exec.return_value = [72 None, self.mock_io("Lustre version: 2.14.0"), self.mock_io()]73 self.assertEqual(1, main.main())74@mock.patch("paramiko.client.SSHClient.connect")75@mock.patch("paramiko.client.SSHClient.exec_command")76class TestMainSSHConnection(BaseTestMain):77 argv = "lfsccm check-ssh-connection"78 def test_check_ssh_connection(self, mock_exec, mock_connect):79 mock_exec.return_value = [80 None, self.mock_io(), self.mock_io()]81 self.assertEqual(0, main.main())...
test_visualisation.py
Source: test_visualisation.py
1"""2Unit tests for visualisation.py module3"""4import unittest5from unittest import mock6from sentiment_comparison import visualisation7class TestVisualisation(unittest.TestCase):8 """ Class for visualisation.py unit tests """9 @mock.patch('sentiment_comparison.visualisation.io')10 @mock.patch('sentiment_comparison.visualisation.plt')11 def test_make_plot__when_number_of_keywords_less_than_four(self, mock_plt, mock_io):12 """ Tests if make_plot function calls methods with appropriate arguments when number of keywords13 less than four """14 sentiments_with_names = {'keyword1': 0.3, 'keyword2': 0.7, 'keyword3': -0.9}15 dpi = 8016 fig_size = (5, 5)17 result = visualisation.make_plot(sentiments_with_names=sentiments_with_names)18 self.assert_called_appropriate_methods(mock_plt, mock_io, dpi, fig_size, result)19 @mock.patch('sentiment_comparison.visualisation.io')20 @mock.patch('sentiment_comparison.visualisation.plt')21 def test_make_plot__when_number_of_keywords_less_than_eight(self, mock_plt, mock_io):22 """ Tests if make_plot function calls methods with appropriate arguments when number of keywords23 greater than four and less than eight """24 sentiments_with_names = {'keyword1': 0.0, 'keyword2': 0.2, 'keyword3': -0.3, 'keyword4': 0.6, 'keyword5': -0.1}25 dpi = 8026 fig_size = (10, 5)27 result = visualisation.make_plot(sentiments_with_names=sentiments_with_names)28 self.assert_called_appropriate_methods(mock_plt, mock_io, dpi, fig_size, result)29 def assert_called_appropriate_methods(self, mock_plt, mock_io, dpi, fig_size, result):30 mock_plt.clf.assert_called_once()31 mock_plt.figure.assert_called_once_with(dpi=dpi, figsize=fig_size)32 mock_plt.bar.assert_called_once()33 mock_plt.title.assert_called_once()34 mock_io.BytesIO.assert_called_once()35 mock_plt.savefig.assert_called_once_with(mock_io.BytesIO.return_value, format='png')36 mock_io.BytesIO.return_value.seek.assert_called_once_with(0)37 self.assertEqual(mock_io.BytesIO.return_value, result[0])38 self.assertEqual(fig_size[0] * dpi, result[1])...
Check out the latest blogs from LambdaTest on this topic:
Before we discuss Scala testing, let us understand the fundamentals of Scala and how this programming language is a preferred choice for your development requirements.The popularity and usage of Scala are rapidly rising, evident by the ever-increasing open positions for Scala developers.
So, now that the first installment of this two fold article has been published (hence you might have an idea of what Agile Testing is not in my opinion), I’ve started feeling the pressure to explain what Agile Testing actually means to me.
Did you know that according to Statista, the number of smartphone users will reach 18.22 billion by 2025? Let’s face it, digital transformation is skyrocketing and will continue to do so. This swamps the mobile app development market with various options and gives rise to the need for the best mobile app testing tools
As a developer, checking the cross browser compatibility of your CSS properties is of utmost importance when building your website. I have often found myself excited to use a CSS feature only to discover that it’s still not supported on all browsers. Even if it is supported, the feature might be experimental and not work consistently across all browsers. Ask any front-end developer about using a CSS feature whose support is still in the experimental phase in most prominent web browsers. ????
The count of mobile users is on a steep rise. According to the research, by 2025, it is expected to reach 7.49 billion users worldwide. 70% of all US digital media time comes from mobile apps, and to your surprise, the average smartphone owner uses ten apps per day and 30 apps each month.
Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.
You could also refer to video tutorials over LambdaTest YouTube channel to get step by step demonstration from industry experts.
Get 100 minutes of automation test minutes FREE!!