def test_04_VariableBinAxis_1D(self): data = ba.IntensityData() data.addAxis(ba.VariableBinAxis("axis0", 10, get_boundaries_flat_in_sin(10, -5.0, 5.0))) fill_data(data) ba.IntensityDataIOFactory.writeOutputData(data, "tmp.int") newdata = ba.IntensityDataIOFactory.readOutputData("tmp.int") self.assertTrue(is_the_same_data(data, newdata))
def test_numpy_array(self): data = ba.IntensityData() data.addAxis("axis0", 10, 0.0, 10.0) data.addAxis("axis1", 20, 0.0, 20.0) data.setAllTo(1) self.assertEqual((20, 10), data.getArray().shape) self.assertEqual((data.totalSum()), numpy.sum(data.getArray()))
def test_01_FixedBinAxis_1D(self): data = ba.IntensityData() data.addAxis(ba.FixedBinAxis("axis0", 10, -1.00000001, 1.0)) fill_data(data) ba.IntensityDataIOFactory.writeOutputData(data, "tmp.int") newdata = ba.IntensityDataIOFactory.readOutputData("tmp.int") self.assertTrue(is_the_same_data(data, newdata))
def test_08_CustomBinAxis_2D(self): data = ba.IntensityData() data.addAxis(ba.CustomBinAxis("axis0", 9, -1.00000001*degree, 1.0*degree)) data.addAxis(ba.CustomBinAxis("axis1", 3, -4.0*degree, 5.0*degree)) fill_data(data) ba.IntensityDataIOFactory.writeOutputData(data, "tmp.int") newdata = ba.IntensityDataIOFactory.readOutputData("tmp.int") self.assertTrue(is_the_same_data(data, newdata))
def test_SaveToTXT(self): data = ba.IntensityData() data.addAxis(ba.FixedBinAxis("axis0", 10, 0.0, 10.0)) data.addAxis(ba.FixedBinAxis("axis1", 5, 0.0, 5.0)) fill_data(data) ba.IntensityDataIOFactory.writeOutputData(data, "tmp.txt") newdata = ba.IntensityDataIOFactory.readOutputData("tmp.txt") self.assertTrue(is_the_same_data(data, newdata))
def test_SaveOutputData_ReadNumpyArray(self): data = ba.IntensityData() data.addAxis(ba.FixedBinAxis("axis0", 10, 0.0, 10.0)) data.addAxis(ba.FixedBinAxis("axis1", 5, 0.0, 5.0)) fill_data(data) ba.IntensityDataIOFactory.writeOutputData(data, "tmp.txt") arr = numpy.loadtxt("tmp.txt") self.assertTrue(numpy.array_equal(data.getArray(),arr))
def test_create_1d_object(self): axis0 = ba.FixedBinAxis("angle", 20, 0.0, 20.) self.assertEqual(20, axis0.size()) self.assertEqual(0.0, axis0.getMin()) self.assertEqual(20.0, axis0.getMax()) data = ba.IntensityData() data.addAxis(axis0) self.assertEqual(20, data.getAllocatedSize()) self.assertEqual(1, data.getRank()) self.assertEqual(0, data.totalSum())
def test_create_2d_object(self): data = ba.IntensityData() data.addAxis("axis0", 10, 0.0, 10.0) data.addAxis("axis1", 20, 0.0, 20.0) self.assertEqual(200, data.getAllocatedSize()) self.assertEqual(2, data.getRank()) self.assertEqual(0, data.totalSum()) data.setAllTo(1.0) self.assertEqual(200.0, data.totalSum()) for i in range(0, data.getAllocatedSize()): data[i] = data[i]*-1.0 self.assertEqual(-200.0, data.totalSum())
def test_SaveToTiff(self): data = ba.IntensityData() data.addAxis(ba.FixedBinAxis("x", 10, 0.0, 10.0)) data.addAxis(ba.FixedBinAxis("y", 5, 0.0, 5.0)) fill_data(data) ba.IntensityDataIOFactory.writeOutputData(data, "tmp.tif") newdata = ba.IntensityDataIOFactory.readOutputData("tmp.tif") self.assertTrue(is_the_same_data(data, newdata)) ba.IntensityDataIOFactory.writeOutputData(data, "tmp.tif.gz") newdata = ba.IntensityDataIOFactory.readOutputData("tmp.tif.gz") self.assertTrue(is_the_same_data(data, newdata)) ba.IntensityDataIOFactory.writeOutputData(data, "tmp.tif.bz2") newdata = ba.IntensityDataIOFactory.readOutputData("tmp.tif.bz2") self.assertTrue(is_the_same_data(data, newdata))
def test_empty_data(self): data = ba.IntensityData() self.assertEqual(1, data.getAllocatedSize()) self.assertEqual(0, data.getRank()) self.assertEqual(0, data.totalSum())