def test_get_weight_tile_for_label_2(self):
img_path = os.path.join(base_path, '../test_data/tiffconnector_1/im/')
label_path = os.path.join(base_path,
'../test_data/tiffconnector_1/labels/')
c = TiffConnector(img_path, label_path)
d = Dataset(c)
d.label_weights[3] = 1.2
img_nr = 2
pos_czxy = np.array((0, 0, 0))
size_czxy = np.array((1, 6, 4))
label_value = 3
mat = d._get_weights_tile(img_nr, pos_czxy, size_czxy, label_value)
val = np.zeros(size_czxy)
print('valshae')
print(val.shape)
val[0, 0, 1] = 1.2
val[0, 4, 1] = 1.2
val[0, 5, 1] = 1.2
pprint('mat')
pprint(mat)
pprint(val)
self.assertTrue((val == mat).all())
def test_get_weight_tile_for_label(self):
img_path = os.path.join(base_path, '../test_data/tiffconnector_1/im/')
label_path = os.path.join(base_path,
'../test_data/tiffconnector_1/labels/')
c = TiffConnector(img_path, label_path)
d = Dataset(c)
img_nr = 2
pos_czxy = (0, 0, 0)
size_czxy = (1, 6, 4)
label_value = 3
mat = d._get_weights_tile(img_nr, pos_czxy, size_czxy, label_value)
val = np.zeros(size_czxy)
val[0, 0, 1] = 1
val[0, 4, 1] = 1
val[0, 5, 1] = 1
np.testing.assert_array_equal(val, mat)
