def test_undoaugment(b, qres):
radius = 1.5
q = transform_StoR(np.array(qres[0], dtype=np.float32), b)
if q.shape == (3, ):
q = q.reshape((1, 3))
q = apply_PBC(q, b)
aug, mapping = augment_coordinates(q, b, radius)
for idx, val in enumerate(aug):
imageid = np.asarray([len(q) + idx], dtype=np.intp)
assert_equal(mapping[idx], undo_augment(imageid, mapping, len(q))[0])
def test_undoaugment(b, qres):
radius = 1.5
q = transform_StoR(np.array(qres[0], dtype=np.float32), b)
if q.shape == (3, ):
q = q.reshape((1, 3))
q = apply_PBC(q, b)
aug, mapping = augment_coordinates(q, b, radius)
for idx, val in enumerate(aug):
imageid = np.asarray([len(q) + idx], dtype=np.int64)
assert_equal(mapping[idx], undo_augment(imageid, mapping, len(q))[0])
def test_augment(b, q, res):
radius = 1.5
q = transform_StoR(np.array(q, dtype=np.float32), b)
if q.shape == (3, ):
q = q.reshape((1, 3))
q = apply_PBC(q, b)
aug, mapping = augment_coordinates(q, b, radius)
if aug.size > 0:
aug = np.sort(aug, axis=0)
else:
aug = list()
if len(res) > 0:
cs = transform_StoR(np.array(res, dtype=np.float32), b)
cs = np.sort(cs, axis=0)
else:
cs = list()
assert_almost_equal(aug, cs, decimal=5)
def test_augment(b, q, res):
radius = 1.5
q = transform_StoR(np.array(q, dtype=np.float32), b)
if q.shape == (3, ):
q = q.reshape((1, 3))
q = apply_PBC(q, b)
aug, mapping = augment_coordinates(q, b, radius)
if aug.size > 0:
aug = np.sort(aug, axis=0)
else:
aug = list()
if len(res) > 0:
cs = transform_StoR(np.array(res, dtype=np.float32), b)
cs = np.sort(cs, axis=0)
else:
cs = list()
assert_almost_equal(aug, cs, decimal=5)