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)
