def test_add_data(self): some_array = DataArray('some_array', 'label', shape=(5, 5)) data_set = DataSet(data_arrays=some_array) data_set.add_data((4, 4), {'some_array': 42}) self.assertEqual(42, some_array[4][4]) data_set.add_data(3, {'some_array': [1, 2, 3, 4, 5]}) self.assertListEqual([1, 2, 3, 4, 5], list(some_array[3]))
def test_add_data_higher_dimensions(self): some_array = DataArray('some_array', 'label', shape=(5, 5, 5, 5)) data_set = DataSet(data_arrays=some_array) data_set.add_data((3, 3, 3, 3), {'some_array': 0.42}) self.assertEqual(0.42, some_array[3][3][3][3]) double_array = [[1, 2, 3, 4, 5], [1, 2, 3, 4, 5], [1, 2, 3, 4, 5], [1, 2, 3, 4, 5], [1, 2, 3, 4, 5]] data_set.add_data((2, 2), {'some_array': double_array}) self.assertTrue(np.array_equal(double_array, some_array[2][2])) self.assertFalse(np.array_equal(double_array, some_array[2][1]))
def test_integrate_with_data_set_io_add_data(self): io_reader, io_writer = MemoryDataSetIOFactory.get_reader_writer_pair() data_set_consumer = DataSet(storage_reader=io_reader) some_array = DataArray('some_array', 'label', shape=(5, 5)) data_set_producer = DataSet(storage_writer=io_writer, data_arrays=some_array) data_set_producer.add_data((0, 0), {'some_array': 42}) data_set_producer.add_data((1, 1), {'some_array': 25}) data_set_consumer.sync_from_storage(-1) self.assertTrue(hasattr(data_set_consumer, 'some_array')) self.assertIsNot(some_array, data_set_consumer.some_array) self.assertEqual(42, data_set_consumer.some_array[0][0]) self.assertEqual(25, data_set_consumer.some_array[1][1])
def test_add_data_set_arrays(self): x_points = np.array(range(0, 2)) y_points = np.array(range(0, 2)) x = DataArray(name="x", label="x-axis", unit="mV", is_setpoint=True, preset_data=x_points) y = DataArray(name="y", label="y-axis", unit="mV", is_setpoint=True, preset_data=np.tile(np.array(y_points), [x.size, 1])) z = DataArray(name="z", label="z-axis", unit="ma", set_arrays=(x, y), preset_data=np.NaN * np.ones( (x_points.size, y_points.size))) data_set = DataSet(data_arrays=z) data_set.add_data((1, 0), {'y': 23}) self.assertEqual(data_set.y[(1, 0)], 23)