Skip to content

Commit

Permalink
test: dispose ImageFrame instances when tests exit
Browse files Browse the repository at this point in the history
  • Loading branch information
@homuler
homuler committed Sep 28, 2021
1 parent a7dbfd7 commit 585c16a
Showing 1 changed file with 80 additions and 75 deletions.
155 changes: 80 additions & 75 deletions Packages/com.github.homuler.mediapipe/Tests/EditMode/Framework/Format/ImageFrameTest.cs
View file
Original file line number Diff line number Diff line change
Expand Up @@ -9,49 +9,50 @@ public class ImageFrameTest {
#region Constructor
[Test, SignalAbort]
public void Ctor_ShouldInstantiateImageFrame_When_CalledWithNoArguments() {
var imageFrame = new ImageFrame();
using (var imageFrame = new ImageFrame()) {
Assert.AreEqual(imageFrame.Format(), ImageFormat.Format.UNKNOWN);
Assert.AreEqual(imageFrame.Width(), 0);
Assert.AreEqual(imageFrame.Height(), 0);
Assert.Throws<FormatException>(() => { imageFrame.ChannelSize(); });
Assert.Throws<FormatException>(() => { imageFrame.NumberOfChannels(); });
Assert.Throws<FormatException>(() => { imageFrame.ByteDepth(); });
Assert.AreEqual(imageFrame.WidthStep(), 0);
Assert.AreEqual(imageFrame.PixelDataSize(), 0);
Assert.Throws<FormatException>(() => { imageFrame.PixelDataSizeStoredContiguously(); });
Assert.True(imageFrame.IsEmpty());
Assert.False(imageFrame.IsContiguous());
Assert.False(imageFrame.IsAligned(16));
Assert.AreEqual(imageFrame.MutablePixelData(), IntPtr.Zero);
}

Assert.AreEqual(imageFrame.Format(), ImageFormat.Format.UNKNOWN);
Assert.AreEqual(imageFrame.Width(), 0);
Assert.AreEqual(imageFrame.Height(), 0);
Assert.Throws<FormatException>(() => { imageFrame.ChannelSize(); });
Assert.Throws<FormatException>(() => { imageFrame.NumberOfChannels(); });
Assert.Throws<FormatException>(() => { imageFrame.ByteDepth(); });
Assert.AreEqual(imageFrame.WidthStep(), 0);
Assert.AreEqual(imageFrame.PixelDataSize(), 0);
Assert.Throws<FormatException>(() => { imageFrame.PixelDataSizeStoredContiguously(); });
Assert.True(imageFrame.IsEmpty());
Assert.False(imageFrame.IsContiguous());
Assert.False(imageFrame.IsAligned(16));
Assert.AreEqual(imageFrame.MutablePixelData(), IntPtr.Zero);
}

[Test]
public void Ctor_ShouldInstantiateImageFrame_When_CalledWithFormat() {
var imageFrame = new ImageFrame(ImageFormat.Format.SBGRA, 640, 480);

Assert.AreEqual(imageFrame.Format(), ImageFormat.Format.SBGRA);
Assert.AreEqual(imageFrame.Width(), 640);
Assert.AreEqual(imageFrame.Height(), 480);
Assert.AreEqual(imageFrame.ChannelSize(), 1);
Assert.AreEqual(imageFrame.NumberOfChannels(), 4);
Assert.AreEqual(imageFrame.ByteDepth(), 1);
Assert.AreEqual(imageFrame.WidthStep(), 640 * 4);
Assert.AreEqual(imageFrame.PixelDataSize(), 640 * 480 * 4);
Assert.AreEqual(imageFrame.PixelDataSizeStoredContiguously(), 640 * 480 * 4);
Assert.False(imageFrame.IsEmpty());
Assert.True(imageFrame.IsContiguous());
Assert.True(imageFrame.IsAligned(16));
Assert.AreNotEqual(imageFrame.MutablePixelData(), IntPtr.Zero);
using (var imageFrame = new ImageFrame(ImageFormat.Format.SBGRA, 640, 480)) {
Assert.AreEqual(imageFrame.Format(), ImageFormat.Format.SBGRA);
Assert.AreEqual(imageFrame.Width(), 640);
Assert.AreEqual(imageFrame.Height(), 480);
Assert.AreEqual(imageFrame.ChannelSize(), 1);
Assert.AreEqual(imageFrame.NumberOfChannels(), 4);
Assert.AreEqual(imageFrame.ByteDepth(), 1);
Assert.AreEqual(imageFrame.WidthStep(), 640 * 4);
Assert.AreEqual(imageFrame.PixelDataSize(), 640 * 480 * 4);
Assert.AreEqual(imageFrame.PixelDataSizeStoredContiguously(), 640 * 480 * 4);
Assert.False(imageFrame.IsEmpty());
Assert.True(imageFrame.IsContiguous());
Assert.True(imageFrame.IsAligned(16));
Assert.AreNotEqual(imageFrame.MutablePixelData(), IntPtr.Zero);
}
}

[Test]
public void Ctor_ShouldInstantiateImageFrame_When_CalledWithFormatAndAlignmentBoundary() {
var imageFrame = new ImageFrame(ImageFormat.Format.GRAY8, 100, 100, 8);

Assert.AreEqual(imageFrame.Width(), 100);
Assert.AreEqual(imageFrame.NumberOfChannels(), 1);
Assert.AreEqual(imageFrame.WidthStep(), 104);
using (var imageFrame = new ImageFrame(ImageFormat.Format.GRAY8, 100, 100, 8)) {
Assert.AreEqual(imageFrame.Width(), 100);
Assert.AreEqual(imageFrame.NumberOfChannels(), 1);
Assert.AreEqual(imageFrame.WidthStep(), 104);
}
}

[Test]
Expand All @@ -63,14 +64,14 @@ public void Ctor_ShouldInstantiateImageFrame_When_CalledWithPixelData() {
};
pixelData.CopyFrom(srcBytes);

var imageFrame = new ImageFrame(ImageFormat.Format.SBGRA, 4, 2, 16, pixelData);

Assert.AreEqual(imageFrame.Width(), 4);
Assert.AreEqual(imageFrame.Height(), 2);
Assert.False(imageFrame.IsEmpty());
using (var imageFrame = new ImageFrame(ImageFormat.Format.SBGRA, 4, 2, 16, pixelData)) {
Assert.AreEqual(imageFrame.Width(), 4);
Assert.AreEqual(imageFrame.Height(), 2);
Assert.False(imageFrame.IsEmpty());

var bytes = imageFrame.CopyToByteBuffer(32);
Assert.IsEmpty(bytes.Where((x, i) => x != srcBytes[i]));
var bytes = imageFrame.CopyToByteBuffer(32);
Assert.IsEmpty(bytes.Where((x, i) => x != srcBytes[i]));
}
}

[Test]
Expand All @@ -82,9 +83,9 @@ public void Ctor_ShouldThrowMediaPipeException_When_CalledWithInvalidArgument()
#region #isDisposed
[Test]
public void isDisposed_ShouldReturnFalse_When_NotDisposedYet() {
var imageFrame = new ImageFrame();

Assert.False(imageFrame.isDisposed);
using (var imageFrame = new ImageFrame()) {
Assert.False(imageFrame.isDisposed);
}
}

[Test]
Expand All @@ -99,72 +100,76 @@ public void isDisposed_ShouldReturnTrue_When_AlreadyDisposed() {
#region #SetToZero
[Test]
public void SetToZero_ShouldSetZeroToAllBytes() {
var imageFrame = new ImageFrame(ImageFormat.Format.GRAY8, 10, 10);
var origBytes = imageFrame.CopyToByteBuffer(100);
Assert.False(origBytes.All((x) => x == 0));

imageFrame.SetToZero();
var bytes = imageFrame.CopyToByteBuffer(100);
Assert.True(bytes.All((x) => x == 0));
using (var imageFrame = new ImageFrame(ImageFormat.Format.GRAY8, 10, 10)) {
var origBytes = imageFrame.CopyToByteBuffer(100);
Assert.False(origBytes.All((x) => x == 0));

imageFrame.SetToZero();
var bytes = imageFrame.CopyToByteBuffer(100);
Assert.True(bytes.All((x) => x == 0));
}
}
#endregion

#region #SetAlignmentPaddingAreas
[Test]
public void SetAlignmentPaddingAreas_ShouldNotThrow() {
var imageFrame = new ImageFrame(ImageFormat.Format.GRAY8, 10, 10, 16);

Assert.DoesNotThrow(() => { imageFrame.SetAlignmentPaddingAreas(); });
using (var imageFrame = new ImageFrame(ImageFormat.Format.GRAY8, 10, 10, 16)) {
Assert.DoesNotThrow(() => { imageFrame.SetAlignmentPaddingAreas(); });
}
}
#endregion

#region CopyToBuffer
[Test]
public void CopyToByteBuffer_ShouldReturnByteArray_When_BufferSizeIsLargeEnough() {
var imageFrame = new ImageFrame(ImageFormat.Format.GRAY8, 10, 10);
var normalBuffer = imageFrame.CopyToByteBuffer(100);
var largeBuffer = imageFrame.CopyToByteBuffer(120);
using (var imageFrame = new ImageFrame(ImageFormat.Format.GRAY8, 10, 10)) {
var normalBuffer = imageFrame.CopyToByteBuffer(100);
var largeBuffer = imageFrame.CopyToByteBuffer(120);

Assert.IsEmpty(normalBuffer.Where((x, i) => x != largeBuffer[i]));
Assert.IsEmpty(normalBuffer.Where((x, i) => x != largeBuffer[i]));
}
}

[Test]
public void CopyToByteBuffer_ShouldThrowException_When_BufferSizeIsTooSmall() {
var imageFrame = new ImageFrame(ImageFormat.Format.GRAY8, 10, 10);

Assert.Throws<MediaPipeException>(() => { imageFrame.CopyToByteBuffer(99); });
using (var imageFrame = new ImageFrame(ImageFormat.Format.GRAY8, 10, 10)) {
Assert.Throws<MediaPipeException>(() => { imageFrame.CopyToByteBuffer(99); });
}
}

[Test]
public void CopyToUshortBuffer_ShouldReturnUshortArray_When_BufferSizeIsLargeEnough() {
var imageFrame = new ImageFrame(ImageFormat.Format.GRAY16, 10, 10);
var normalBuffer = imageFrame.CopyToUshortBuffer(100);
var largeBuffer = imageFrame.CopyToUshortBuffer(120);
using (var imageFrame = new ImageFrame(ImageFormat.Format.GRAY16, 10, 10)) {
var normalBuffer = imageFrame.CopyToUshortBuffer(100);
var largeBuffer = imageFrame.CopyToUshortBuffer(120);

Assert.IsEmpty(normalBuffer.Where((x, i) => x != largeBuffer[i]));
Assert.IsEmpty(normalBuffer.Where((x, i) => x != largeBuffer[i]));
}
}

[Test]
public void CopyToUshortBuffer_ShouldThrowException_When_BufferSizeIsTooSmall() {
var imageFrame = new ImageFrame(ImageFormat.Format.GRAY16, 10, 10);

Assert.Throws<MediaPipeException>(() => { imageFrame.CopyToUshortBuffer(99); });
using (var imageFrame = new ImageFrame(ImageFormat.Format.GRAY16, 10, 10)) {
Assert.Throws<MediaPipeException>(() => { imageFrame.CopyToUshortBuffer(99); });
}
}

[Test]
public void CopyToFloatBuffer_ShouldReturnFloatArray_When_BufferSizeIsLargeEnough() {
var imageFrame = new ImageFrame(ImageFormat.Format.VEC32F1, 10, 10);
var normalBuffer = imageFrame.CopyToFloatBuffer(100);
var largeBuffer = imageFrame.CopyToFloatBuffer(120);
using (var imageFrame = new ImageFrame(ImageFormat.Format.VEC32F1, 10, 10)) {
var normalBuffer = imageFrame.CopyToFloatBuffer(100);
var largeBuffer = imageFrame.CopyToFloatBuffer(120);

Assert.IsEmpty(normalBuffer.Where((x, i) => Math.Abs(x - largeBuffer[i]) > 1e-9));
Assert.IsEmpty(normalBuffer.Where((x, i) => Math.Abs(x - largeBuffer[i]) > 1e-9));
}
}

[Test]
public void CopyToFloatBuffer_ShouldThrowException_When_BufferSizeIsTooSmall() {
var imageFrame = new ImageFrame(ImageFormat.Format.VEC32F1, 10, 10);

Assert.Throws<MediaPipeException>(() => { imageFrame.CopyToFloatBuffer(99); });
using (var imageFrame = new ImageFrame(ImageFormat.Format.VEC32F1, 10, 10)) {
Assert.Throws<MediaPipeException>(() => { imageFrame.CopyToFloatBuffer(99); });
}
}
#endregion
}
Expand Down

0 comments on commit 585c16a

Please sign in to comment.