Class Test_AABB_Intersection

tests/plib/test_utils.py:9–55

class Test_AABB_Intersection(unittest.TestCase):
    def test_1(self):
        N = 10000
        stime = timer()
        for _ in range(N):
            out_dict = utils.ray_aabb_intersection(
                ray_origin=torch.zeros(3),
                ray_direction=torch.tensor([0,0,1.]),
                bbox_min_bounds=torch.tensor([0,0,2.]),
                bbox_max_bounds=torch.tensor([10, 10, 10.]),
            )
        etime = timer()
        print(f'v1 takes {(etime - stime) / N} secs per test')
        assert out_dict['is_intersected']
        assert np.isclose(out_dict['t_near'].detach().cpu().numpy(), 2.)
        assert np.isclose(out_dict['t_far'].detach().cpu().numpy(), 10.)


    def test_2(self):
        N = 10000
        stime = timer()
        for _ in range(N):
            out_dict = utils.ray_aabb_intersection_2(
                ray_origin=torch.zeros(3),
                ray_direction=torch.tensor([0,0,1.]),
                bbox_min_bounds=torch.tensor([0,0,2.]),
                bbox_max_bounds=torch.tensor([10, 10, 10.]),
            )
        etime = timer()
        print(f'v2 takes {(etime - stime) / N} secs per test')
        assert out_dict['is_intersected']
        assert np.isclose(out_dict['t_near'].detach().cpu().numpy(), 2.)
        assert np.isclose(out_dict['t_far'].detach().cpu().numpy(), 10.)

    def test_3(self):
        N = 10000
        stime = timer()
        for _ in range(N):
            out_dict = utils.ray_aabb_intersection(
                ray_origin=torch.zeros(3),
                ray_direction=torch.tensor([0,0,1.]),
                bbox_min_bounds=torch.tensor([1, 1, 2.]),
                bbox_max_bounds=torch.tensor([10, 10, 10.]),
            )
        etime = timer()
        print(f'v1 takes {(etime - stime) / N} secs per test')
        assert out_dict['is_intersected'] == False