def test_OpNpyFileReaderNPZnamed(self):
# Now read back our test data using an OpNpyFileReader operator
npyReader = OpNpyFileReader(graph=self.graph)
try:
for internalPath, referenceData in zip(["data_A", "data_B"], [self.testData, self.testDataB]):
npyReader.InternalPath.setValue("data_B")
npyReader.FileName.setValue(self.testDataFilePathNPZnamed)
# Read the entire file and verify the contents
b = npyReader.Output[:].wait()
assert b.shape == self.testDataB.shape
assert npyReader.Output.meta.dtype == self.testDataB.dtype
numpy.testing.assert_almost_equal(b, self.testDataB)
finally:
npyReader.cleanUp()
def test_OpNpyFileReaderNPZnamed(self):
# Now read back our test data using an OpNpyFileReader operator
npyReader = OpNpyFileReader(graph=self.graph)
try:
for internalPath, referenceData in zip(
["data_A", "data_B"], [self.testData, self.testDataB]):
npyReader.InternalPath.setValue("data_B")
npyReader.FileName.setValue(self.testDataFilePathNPZnamed)
# Read the entire file and verify the contents
b = npyReader.Output[:].wait()
assert b.shape == self.testDataB.shape
assert npyReader.Output.meta.dtype == self.testDataB.dtype
numpy.testing.assert_almost_equal(b, self.testDataB)
finally:
npyReader.cleanUp()
def test_OpNpyFileReader(self):
# Now read back our test data using an OpNpyFileReader operator
npyReader = OpNpyFileReader(graph=self.graph)
npyReader.FileName.setValue(self.testDataFilePath)
# Read the entire file and verify the contents
a = npyReader.Output[:].wait()
assert a.shape == (10,11) # OpNpyReader automatically added a channel axis
assert npyReader.Output.meta.dtype == self.testData.dtype
# Why doesn't this work? Numpy bug?
# cmp = ( a == self.testData )
# assert cmp.all()
# Check each of the values
for i in range(10):
for j in range(11):
assert a[i,j] == self.testData[i,j]
npyReader.cleanUp()
def test_OpNpyFileReader(self):
# Now read back our test data using an OpNpyFileReader operator
npyReader = OpNpyFileReader(graph=self.graph)
npyReader.FileName.setValue(self.testDataFilePath)
# Read the entire file and verify the contents
a = npyReader.Output[:].wait()
assert a.shape == (10, 11
) # OpNpyReader automatically added a channel axis
assert npyReader.Output.meta.dtype == self.testData.dtype
# Why doesn't this work? Numpy bug?
# cmp = ( a == self.testData )
# assert cmp.all()
# Check each of the values
for i in range(10):
for j in range(11):
assert a[i, j] == self.testData[i, j]
npyReader.cleanUp()