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)