/*
* This file is part of jHDF. A pure Java library for accessing HDF5 files.
*
* http://jhdf.io
*
* Copyright (c) 2020 James Mudd
*
* MIT License see 'LICENSE' file
*/
package io.jhdf;
import io.jhdf.api.Dataset;
import io.jhdf.dataset.DatasetBase;
import io.jhdf.object.message.DataLayout;
import org.junit.jupiter.api.AfterAll;
import org.junit.jupiter.api.BeforeAll;
import org.junit.jupiter.api.Test;
import java.nio.ByteBuffer;
import java.nio.DoubleBuffer;
import java.nio.FloatBuffer;
import java.nio.IntBuffer;
import java.nio.ShortBuffer;
import static io.jhdf.TestUtils.loadTestHdfFile;
import static org.hamcrest.MatcherAssert.assertThat;
import static org.hamcrest.Matchers.equalTo;
import static org.hamcrest.Matchers.is;
import static org.hamcrest.Matchers.notNullValue;
import static org.junit.jupiter.api.Assertions.assertArrayEquals;
class DatasetImplTest {
private static final String INT8_PATH = "/datasets_group/int/int8";
private static final String INT16_PATH = "/datasets_group/int/int16";
private static final String INT32_PATH = "/datasets_group/int/int32";
private static final String FLOAT64_PATH = "/datasets_group/float/float64";
private static final String FLOAT32_PATH = "/datasets_group/float/float32";
private static final String HDF5_TEST_FILE_NAME = "test_file.hdf5";
// These are the "correct" contents of the datasets i.e need to match what was
// written by h5py
private static final byte[] REFERENCE_BYTE_DATA = new byte[] { -10, -9, -8, -7, -6, -5, -4, -3, -2, -1, 0, 1, 2, 3,
4, 5, 6, 7, 8, 9, 10 };
private static final short[] REFERENCE_SHORT_DATA = new short[] { -10, -9, -8, -7, -6, -5, -4, -3, -2, -1, 0, 1, 2,
3, 4, 5, 6, 7, 8, 9, 10 };
private static final int[] REFERENCE_INT_DATA = new int[] { -10, -9, -8, -7, -6, -5, -4, -3, -2, -1, 0, 1, 2, 3, 4,
5, 6, 7, 8, 9, 10 };
private static final double[] REFERENCE_DOUBLE_DATA = new double[] { -10, -9, -8, -7, -6, -5, -4, -3, -2, -1, 0, 1,
2, 3, 4, 5, 6, 7, 8, 9, 10 };
private static final float[] REFERENCE_FLOAT_DATA = new float[] { -10, -9, -8, -7, -6, -5, -4, -3, -2, -1, 0, 1, 2,
3, 4, 5, 6, 7, 8, 9, 10 };
// Tolerance to use when comparing float values
private static final float FLOAT_DELTA = 0.000000001f;
private static final double DOUBLE_DELTA = 0.000000001;
private static HdfFile hdfFile;
@BeforeAll
static void beforeAll() throws Exception {
hdfFile = loadTestHdfFile(HDF5_TEST_FILE_NAME);
}
@AfterAll
static void afterAll() {
hdfFile.close();
}
@Test
void testByteDatasetBuffer() {
DatasetBase dataset = (DatasetBase) hdfFile.getByPath(INT8_PATH);
// We know this is a float dataset so get the float buffer
ByteBuffer buffer = dataset.getDataBuffer();
assertThat(buffer, is(notNullValue()));
// Check the size of the buffer is correct, i.e the number of points in this
// data set
assertThat(buffer.limit(), is(equalTo(21)));
byte[] data = new byte[21];
// read the data into the array
buffer.get(data);
// check the data is correct
assertThat(data, is(equalTo(REFERENCE_BYTE_DATA)));
}
@Test
void testShortDatasetBuffer() {
DatasetBase dataset = (DatasetBase) hdfFile.getByPath(INT16_PATH);
// We know this is a short dataset so get the short buffer
ShortBuffer buffer = dataset.getDataBuffer().asShortBuffer();
assertThat(buffer, is(notNullValue()));
// Check the size of the buffer is correct, i.e the number of points in this
// data set
assertThat(buffer.limit(), is(equalTo(21)));
short[] data = new short[21];
// read the data into the array
buffer.get(data);
// check the data is correct
assertArrayEquals(REFERENCE_SHORT_DATA, data);
}
@Test
void testIntDatasetBuffer() {
DatasetBase dataset = (DatasetBase) hdfFile.getByPath(INT32_PATH);
// We know this is a int dataset so get the int buffer
IntBuffer buffer = dataset.getDataBuffer().asIntBuffer();
assertThat(buffer, is(notNullValue()));
// Check the size of the buffer is correct, i.e the number of points in this
// data set
assertThat(buffer.limit(), is(equalTo(21)));
int[] data = new int[21];
// read the data into the array
buffer.get(data);
// check the data is correct
assertArrayEquals(REFERENCE_INT_DATA, data);
}
@Test
void testFloatDatasetBuffer() {
DatasetBase dataset = (DatasetBase) hdfFile.getByPath(FLOAT32_PATH);
// We know this is a float dataset so get the float buffer
FloatBuffer buffer = dataset.getDataBuffer().asFloatBuffer();
assertThat(buffer, is(notNullValue()));
// Check the size of the buffer is correct, i.e the number of points in this
// data set
assertThat(buffer.limit(), is(equalTo(21)));
float[] data = new float[21];
// read the data into the array
buffer.get(data);
// check the data is correct
assertArrayEquals(REFERENCE_FLOAT_DATA, data, FLOAT_DELTA);
}
@Test
void testDoubleDatasetBuffer() {
DatasetBase dataset = (DatasetBase) hdfFile.getByPath(FLOAT64_PATH);
// We know this is a double dataset so get the double buffer
DoubleBuffer buffer = dataset.getDataBuffer().asDoubleBuffer();
assertThat(buffer, is(notNullValue()));
// Check the size of the buffer is correct, i.e the number of points in this
// data set
assertThat(buffer.limit(), is(equalTo(21)));
double[] data = new double[21];
// read the data into the array
buffer.get(data);
// check the data is correct
assertArrayEquals(REFERENCE_DOUBLE_DATA, data, DOUBLE_DELTA);
}
@Test
void testByteDatasetSize() {
Dataset dataset = (Dataset) hdfFile.getByPath(INT8_PATH);
assertThat(dataset.getSize(), is(equalTo(21L)));
}
@Test
void testByteDatasetDiskSize() {
Dataset dataset = (Dataset) hdfFile.getByPath(INT8_PATH);
assertThat(dataset.getDiskSize(), is(equalTo(21L)));
}
@Test
void testShortDatasetSize() {
Dataset dataset = (Dataset) hdfFile.getByPath(INT16_PATH);
assertThat(dataset.getSize(), is(equalTo(21L)));
}
@Test
void testShortDatasetDiskSize() {
Dataset dataset = (Dataset) hdfFile.getByPath(INT16_PATH);
assertThat(dataset.getDiskSize(), is(equalTo(42L)));
}
@Test
void testInt32DatasetSize() {
Dataset dataset = (Dataset) hdfFile.getByPath(INT32_PATH);
assertThat(dataset.getSize(), is(equalTo(21L)));
}
@Test
void testInt32DatasetDiskSize() {
Dataset dataset = (Dataset) hdfFile.getByPath(INT32_PATH);
assertThat(dataset.getDiskSize(), is(equalTo(84L)));
}
@Test
void testInt32DatasetDimensions() {
Dataset dataset = (Dataset) hdfFile.getByPath(INT32_PATH);
assertThat(dataset.getDimensions(), is(equalTo(new int[] { 21 })));
}
@Test
void testInt32DatasetMaxSizes() {
Dataset dataset = (Dataset) hdfFile.getByPath(INT32_PATH);
assertThat(dataset.getMaxSize(), is(equalTo(new int[] { 21 })));
}
@Test
void testInt32DatasetDataLayout() {
Dataset dataset = (Dataset) hdfFile.getByPath(INT32_PATH);
assertThat(dataset.getDataLayout(), is(DataLayout.CONTIGUOUS));
}
@Test
void testInt32DatasetJavaType() {
Dataset dataset = (Dataset) hdfFile.getByPath(INT32_PATH);
assertThat(dataset.getJavaType(), is(equalTo(int.class)));
}
@Test
void testInt32GetData() {
Dataset dataset = (Dataset) hdfFile.getByPath(INT32_PATH);
assertThat(dataset.getData(), is(equalTo(REFERENCE_INT_DATA)));
}
@Test
void testFloat32DatasetJavaType() {
Dataset dataset = (Dataset) hdfFile.getByPath(FLOAT32_PATH);
assertThat(dataset.getJavaType(), is(equalTo(float.class)));
}
@Test
void testFloat32DatasetGetData() {
Dataset dataset = (Dataset) hdfFile.getByPath(FLOAT32_PATH);
assertThat(dataset.getData(), is(equalTo(REFERENCE_FLOAT_DATA)));
}
@Test
void testFloat64DatasetGetData() {
Dataset dataset = (Dataset) hdfFile.getByPath(FLOAT64_PATH);
assertThat(dataset.getData(), is(equalTo(REFERENCE_DOUBLE_DATA)));
}
}
