def test_makedataset2d_shape_measuredata_second_nok(self): p1 = ManualParameter('dummy1') p2 = ManualParameter('dummy2') x = p1[0:10:1] y = p2[0:4:1] measure_names = ['measured1', 'measured2'] preset_data = [ np.ones((len(x), len(y))), np.ones((len(x) + 1, len(y))) ] with patch('sys.stdout', new_callable=io.StringIO) as mock_stdout: stream_handler = logging.StreamHandler(sys.stdout) logger.addHandler(stream_handler) _ = qtt.data.makeDataSet2D(x, y, measure_names, preset_data=preset_data, return_names=False) # Verify warning print_string = mock_stdout.getvalue() self.assertRegex( print_string, 'Shape of measured data .* does not match setpoint shape .*') logger.removeHandler(stream_handler)
def test_dataset_no_measured_data(self): with patch('sys.stdout', new_callable=io.StringIO) as mock_stdout: stream_handler = logging.StreamHandler(sys.stdout) logger.addHandler(stream_handler) dataset = qtt.data.makeDataSet1Dplain('x', [0, 1], 'y', []) # Verify warning print_string = mock_stdout.getvalue() self.assertEqual( print_string, 'Shape of measured data (0,) does not match setpoint shape (2,)\n' ) logger.removeHandler(stream_handler) with patch('sys.stdout', new_callable=io.StringIO) as mock_stdout: stream_handler = logging.StreamHandler(sys.stdout) logger.addHandler(stream_handler) dataset = qtt.data.makeDataSet1Dplain('x', [0, 1], 'y', None) # Verify in this case we have no warning print_string = mock_stdout.getvalue() self.assertEqual(print_string, '') logger.removeHandler(stream_handler)
def test_dataset2dplain_shape_measuredata_nok(self): with patch('sys.stdout', new_callable=io.StringIO) as mock_stdout: stream_handler = logging.StreamHandler(sys.stdout) logger.addHandler(stream_handler) _ = qtt.data.makeDataSet2Dplain('horizontal', [0., 1, 2, 3], 'vertical', [0, 1, 2.], 'z', np.arange(3 * 5).reshape((3, 5)), xunit='mV', yunit='Hz', zunit='A') # Verify warning print_string = mock_stdout.getvalue() self.assertRegex(print_string, 'Shape of measured data .* does not match setpoint shape .*' ) logger.removeHandler(stream_handler)
def test_dataset1dplain_other_shape(self): x = [1, 2, 3] y = np.array([1, 2, 3]).reshape((3, 1)) with patch('sys.stdout', new_callable=io.StringIO) as mock_stdout: stream_handler = logging.StreamHandler(sys.stdout) logger.addHandler(stream_handler) data_set = qtt.data.makeDataSet1Dplain('x', x, 'y', y) self.assertTrue(np.array_equal(data_set.x, x)) self.assertTrue(np.array_equal(data_set.y, y)) # Verify warning print_string = mock_stdout.getvalue() self.assertRegex(print_string, 'Shape of measured data .* does not match setpoint shape .*' ) logger.removeHandler(stream_handler)
def test_makedataset1dplain_shape_measuredata_list_nok2(self): x = np.arange(0, 10) y = np.arange(1, 11) with patch('sys.stdout', new_callable=io.StringIO) as mock_stdout: stream_handler = logging.StreamHandler(sys.stdout) logger.addHandler(stream_handler) data_set = qtt.data.makeDataSet1Dplain('x', x, 'y1', [y, y]) self.assertTrue(np.array_equal(data_set.x, x)) self.assertTrue(np.array_equal(data_set.y1, [y, y])) # Verify warning print_string = mock_stdout.getvalue() self.assertRegex(print_string, 'Shape of measured data .* does not match setpoint shape .*' ) logger.removeHandler(stream_handler)
def test_makedataset1d_shape_measuredata_nok(self): dummy_parameter = ManualParameter('dummy') x = dummy_parameter[0:10:1] y = [np.arange(len(x)+1)] with patch('sys.stdout', new_callable=io.StringIO) as mock_stdout: stream_handler = logging.StreamHandler(sys.stdout) logger.addHandler(stream_handler) data_set = qtt.data.makeDataSet1D(x, 'y', y, return_names=False) self.assertTrue(np.array_equal(data_set.dummy, x)) self.assertTrue(np.array_equal(data_set.y, y)) # Verify warning print_string = mock_stdout.getvalue() self.assertRegex(print_string, 'Shape of measured data .* does not match setpoint shape .*' ) logger.removeHandler(stream_handler)