app/model/point-trajectory.group.spec.ts

0001 // point-trajectory.component.spec.ts
0002 import { PointTrajectoryGroup, PointTrajectoryGroupFactory } from './point-trajectory.group';
0003 import { getEventGroupFactory } from './event-group';
0004 import { initGroupFactories } from "./default-group-init";
0005
0006
0007 describe('PointTrajectoryGroup', () => {
0008     beforeEach(() => {
0009         initGroupFactories();
0010     });
0011
0012     it('should be registered in the component factory registry', () => {
0013         const factory = getEventGroupFactory(PointTrajectoryGroup.type);
0014         expect(factory).toBeDefined();
0015         expect(factory instanceof PointTrajectoryGroupFactory).toBe(true);
0016     });
0017
0018     it('should create a component from a valid Dex object', () => {
0019         // Prepare a mock DEX object
0020         const dexObj = {
0021             name: 'CentralTrackSegments',
0022             type: 'PointTrajectory',
0023             origin: ['edm4eic::TrackPoint', 'edm4eic::TrackSegmentData'],
0024             paramColumns: ['theta', 'phi', 'qOverP', 'charge', 'pdg'],
0025             pointColumns: ['x', 'y', 'z', 't', 'dx', 'dy', 'dz', 'dt'],
0026             trajectories: [
0027                 {
0028                     points: [
0029                         [0, 0, 0, 100, 0.1, 0.1, 0.1, 0.0],
0030                         [10, 20, 30, 120, 0.2, 0.2, 0.2, 0.1],
0031                     ],
0032                     params: [1.57, 3.14, -0.0005, -1, 11]
0033                 },
0034                 {
0035                     points: [],
0036                     params: []
0037                 }
0038             ]
0039         };
0040
0041         // Retrieve the factory and create the component
0042         const factory = getEventGroupFactory(PointTrajectoryGroup.type);
0043         expect(factory).toBeTruthy();
0044
0045         const component = factory?.fromDexObject(dexObj) as PointTrajectoryGroup;
0046         expect(component).toBeDefined();
0047         expect(component.name).toBe('CentralTrackSegments');
0048         expect(component.paramColumns).toEqual(['theta', 'phi', 'qOverP', 'charge', 'pdg']);
0049         expect(component.pointColumns).toEqual(['x', 'y', 'z', 't', 'dx', 'dy', 'dz', 'dt']);
0050
0051         // Lines
0052         expect(component.trajectories.length).toBe(2);
0053         // First line
0054         const trajectory1 = component.trajectories[0];
0055         expect(trajectory1.points.length).toBe(2);
0056         expect(trajectory1.params).toEqual([1.57, 3.14, -0.0005, -1, 11]);
0057         // Second line
0058         const trajectory2 = component.trajectories[1];
0059         expect(trajectory2.points.length).toBe(0);
0060         expect(trajectory2.params.length).toBe(0);
0061     });
0062
0063     it('should serialize back to Dex format via toDexObject()', () => {
0064         // Create a component manually
0065         const component = new PointTrajectoryGroup('MyTrajectory', 'testOrigin');
0066         component.paramColumns = ['alpha', 'beta'];
0067         component.pointColumns = ['x', 'y', 'z', 't'];
0068         component.trajectories = [
0069             {
0070                 points: [[1, 2, 3, 10], [4, 5, 6, 20]],
0071                 params: [42, 99]
0072             }
0073         ];
0074
0075         // Convert to Dex
0076         const dex = component.toDexObject();
0077
0078         // Check top-level
0079         expect(dex.name).toBe('MyTrajectory');
0080         expect(dex.type).toBe('PointTrajectory');
0081         expect(dex.origin).toBe('testOrigin');
0082         expect(dex.paramColumns).toEqual(['alpha', 'beta']);
0083         expect(dex.pointColumns).toEqual(['x', 'y', 'z', 't']);
0084
0085         // Check lines
0086         expect(dex.trajectories.length).toBe(1);
0087         expect(dex.trajectories[0].points.length).toBe(2);
0088         expect(dex.trajectories[0].points[0]).toEqual([1, 2, 3, 10]);
0089         expect(dex.trajectories[0].params).toEqual([42, 99]);
0090     });
0091
0092 });