TmMolStar/src/reader/spec/gro.spec.ts

/*
 * Copyright (c) 2017 molio contributors, licensed under MIT, See LICENSE file for more info.
 *
 * @author Alexander Rose <alexander.rose@weirdbyte.de>
 */

import { parse } from '../gro'
// import { Category } from '../../relational/category'

const groString = `MD of 2 waters, t= 4.2
    6
    1WATER  OW1    1   0.126   1.624   1.679  0.1227 -0.0580  0.0434
    1WATER  HW2    2   0.190   1.661   1.747  0.8085  0.3191 -0.7791
    1WATER  HW3    3   0.177   1.568   1.613 -0.9045 -2.6469  1.3180
    2WATER  OW1    4   1.275   0.053   0.622  0.2519  0.3140 -0.1734
    2WATER  HW2    5   1.337   0.002   0.680 -1.0641 -1.1349  0.0257
    2WATER  HW3    6   1.326   0.120   0.568  1.9427 -0.8216 -0.0244
   1.82060   1.82060   1.82060`

const groStringHighPrecision = `Generated by trjconv : 2168 system t=  15.00000
    3
    1ETH     C1    1   2.735383   2.672010   1.450194  0.2345 -0.1622 0.2097
    1ETH    H11    2   0.015804   2.716597   1.460588  0.8528 -0.7984 0.6605
    1ETH    H12    3   2.744822   2.565544   1.409227 -2.3812  2.8618 1.8101
   1.82060   1.82060   1.82060`

describe('gro reader', () => {
    it('basic', () => {
        const parsed = parse(groString)

        if (parsed.isError) {
            console.log(parsed)
        } else {
            const groFile = parsed.result

            const header = groFile.blocks[0].getCategory('header')
            if (header) {
                expect(header.columnNames).toEqual(['title', 'timeInPs', 'numberOfAtoms', 'boxX', 'boxY', 'boxZ'])

                expect(header.getColumn('title').getString(0)).toBe('MD of 2 waters')
                expect(header.getColumn('timeInPs').getFloat(0)).toBeCloseTo(4.2)
                expect(header.getColumn('numberOfAtoms').getInteger(0)).toBe(6)

                expect(header.getColumn('boxX').getFloat(0)).toBeCloseTo(1.82060)
                expect(header.getColumn('boxY').getFloat(0)).toBeCloseTo(1.82060)
                expect(header.getColumn('boxZ').getFloat(0)).toBeCloseTo(1.82060)
            } else {
                console.error('no header')
            }
        }
    })

    it('high precision', () => {
        const parsed = parse(groStringHighPrecision)

        if (parsed.isError) {
            console.log(parsed)
        } else {
            const groFile = parsed.result

            const header = groFile.blocks[0].getCategory('header')
            if (header) {
                expect(header.columnNames).toEqual(['title', 'timeInPs', 'numberOfAtoms', 'boxX', 'boxY', 'boxZ'])

                expect(header.getColumn('title').getString(0)).toBe('Generated by trjconv : 2168 system')
                expect(header.getColumn('timeInPs').getFloat(0)).toBeCloseTo(15)
                expect(header.getColumn('numberOfAtoms').getInteger(0)).toBe(3)

                expect(header.getColumn('boxX').getFloat(0)).toBeCloseTo(1.82060)
                expect(header.getColumn('boxY').getFloat(0)).toBeCloseTo(1.82060)
                expect(header.getColumn('boxZ').getFloat(0)).toBeCloseTo(1.82060)
            } else {
                console.error('no header')
            }
        }
    })
});