Java/src/test/java/com/thealgorithms/physics/ElasticCollision2DTest.java at master · SwaatiR/Java · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
package com.thealgorithms.physics;

import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertTrue;

import org.junit.jupiter.api.Test;

class ElasticCollision2DTest {

    @Test
    void testEqualMassHeadOnCollision() {
        ElasticCollision2D.Body a = new ElasticCollision2D.Body(0, 0, 1, 0, 1.0, 0.5);
        ElasticCollision2D.Body b = new ElasticCollision2D.Body(1, 0, -1, 0, 1.0, 0.5);

        ElasticCollision2D.resolveCollision(a, b);

        assertEquals(-1.0, a.vx, 1e-6);
        assertEquals(0.0, a.vy, 1e-6);
        assertEquals(1.0, b.vx, 1e-6);
        assertEquals(0.0, b.vy, 1e-6);
    }

    @Test
    void testUnequalMassHeadOnCollision() {
        ElasticCollision2D.Body a = new ElasticCollision2D.Body(0, 0, 2, 0, 2.0, 0.5);
        ElasticCollision2D.Body b = new ElasticCollision2D.Body(1, 0, -1, 0, 1.0, 0.5);

        ElasticCollision2D.resolveCollision(a, b);

        // 1D head-on collision results
        assertEquals(0.0, a.vx, 1e-6);
        assertEquals(0.0, a.vy, 1e-6);
        assertEquals(3.0, b.vx, 1e-6);
        assertEquals(0.0, b.vy, 1e-6);
    }

    @Test
    void testMovingApartNoCollision() {
        ElasticCollision2D.Body a = new ElasticCollision2D.Body(0, 0, -1, 0, 1.0, 0.5);
        ElasticCollision2D.Body b = new ElasticCollision2D.Body(1, 0, 1, 0, 1.0, 0.5);

        ElasticCollision2D.resolveCollision(a, b);

        assertEquals(-1.0, a.vx, 1e-6);
        assertEquals(0.0, a.vy, 1e-6);
        assertEquals(1.0, b.vx, 1e-6);
        assertEquals(0.0, b.vy, 1e-6);
    }

    @Test
    void testGlancingCollision() {
        ElasticCollision2D.Body a = new ElasticCollision2D.Body(0, 0, 1, 1, 1.0, 0.5);
        ElasticCollision2D.Body b = new ElasticCollision2D.Body(1, 1, -1, -0.5, 1.0, 0.5);

        ElasticCollision2D.resolveCollision(a, b);

        // Ensure relative velocity along normal is reversed
        double nx = (b.x - a.x) / Math.hypot(b.x - a.x, b.y - a.y);
        double ny = (b.y - a.y) / Math.hypot(b.x - a.x, b.y - a.y);
        double relVelAfter = (b.vx - a.vx) * nx + (b.vy - a.vy) * ny;

        assertTrue(relVelAfter > 0);
    }

    @Test
    void testOverlappingBodies() {
        ElasticCollision2D.Body a = new ElasticCollision2D.Body(0, 0, 1, 0, 1.0, 0.5);
        ElasticCollision2D.Body b = new ElasticCollision2D.Body(0, 0, -1, 0, 1.0, 0.5);

        ElasticCollision2D.resolveCollision(a, b);

        // Should not crash, velocities may remain unchanged
        assertEquals(1.0, a.vx, 1e-6);
        assertEquals(0.0, a.vy, 1e-6);
        assertEquals(-1.0, b.vx, 1e-6);
        assertEquals(0.0, b.vy, 1e-6);
    }

    @Test
    void testStationaryBodyHit() {
        ElasticCollision2D.Body a = new ElasticCollision2D.Body(0, 0, 2, 0, 1.0, 0.5);
        ElasticCollision2D.Body b = new ElasticCollision2D.Body(1, 0, 0, 0, 1.0, 0.5);

        ElasticCollision2D.resolveCollision(a, b);

        assertEquals(0.0, a.vx, 1e-6);
        assertEquals(0.0, a.vy, 1e-6);
        assertEquals(2.0, b.vx, 1e-6);
        assertEquals(0.0, b.vy, 1e-6);
    }
}