OpenCloud
/
PeerTube
mirror of https://github.com/Chocobozzz/PeerTube
1
0
Fork 0
Code Issues Releases Wiki Activity

chore(test): refacto timestamed spec

pull/6303/head
lutangar 2024-04-29 17:34:19 +02:00
parent 380bd49140
commit c683452aee
1 changed files with 18 additions and 26 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

44
packages/tests/src/transcription/whisper/transcriber/timestamped-transcriber.spec.ts
View File

@ -2,10 +2,9 @@
import { expect, config } from 'chai'
import { createLogger } from 'winston'
import { join } from 'path'
import { existsSync } from 'node:fs'
import { mkdir, readFile, rm } from 'node:fs/promises'
import { mkdir, rm } from 'node:fs/promises'
import { buildAbsoluteFixturePath, root } from '@peertube/peertube-node-utils'
import { OpenaiTranscriber, WhisperTimestampedTranscriber } from '@peertube/peertube-transcription'
import { OpenaiTranscriber, WhisperTimestampedTranscriber, TranscriptFile } from '@peertube/peertube-transcription'
config.truncateThreshold = 0
@ -29,24 +28,20 @@ describe('Linto timestamped Whisper transcriber', function () {
await mkdir(transcriptDirectory, { recursive: true })
})
it('Should transcribe a media file and produce transcript file in th `vtt` format by default', async function () {
it('Should transcribe a media file and produce a transcript file in `vtt` with a ms precision', async function () {
const transcript = await transcriber.transcribe(
shortVideoPath,
{ name: 'tiny' },
'fr',
'vtt'
'fr'
)
expect(transcript).to.deep.equals({
expect(transcript.equals(new TranscriptFile({
path: join(transcriptDirectory, 'video_short.vtt'),
language: 'fr',
format: 'vtt'
})
}))).to.be.true
expect(existsSync(transcript.path), `Transcript file ${transcript.path} doesn't exist.`).to.be.true
// Whisper timestamped should produce a transcript with micro seconds precisions.
expect(await readFile(transcript.path, 'utf8')).to.equal(
expect(await transcript.read()).to.equals(
`WEBVTT
00:02.480 --> 00:02.500
@ -56,16 +51,15 @@ you
)
})
it('May produce a transcript file in the `srt` format', async function () {
it('May produce a transcript file in the `srt` format with a ms precision', async function () {
const transcript = await transcriber.transcribe(shortVideoPath, { name: 'tiny' }, 'en', 'srt')
expect(transcript).to.deep.equals({
expect(transcript.equals(new TranscriptFile({
path: join(transcriptDirectory, 'video_short.srt'),
language: 'en',
format: 'srt'
})
}))).to.be.true
expect(existsSync(transcript.path), `Transcript file ${transcript.path} doesn't exist.`).to.be.true
expect(await readFile(transcript.path, 'utf8')).to.equal(
expect(await transcript.read()).to.equals(
`1
00:00:02,480 --> 00:00:02,500
you
@ -76,28 +70,26 @@ you
it('May produce a transcript file in `txt` format', async function () {
const transcript = await transcriber.transcribe(shortVideoPath, { name: 'tiny' }, 'en', 'txt')
expect(transcript).to.deep.equals({
expect(transcript.equals(new TranscriptFile({
path: join(transcriptDirectory, 'video_short.txt'),
language: 'en',
format: 'txt'
})
}))).to.be.true
expect(existsSync(transcript.path), `Transcript file ${transcript.path} doesn't exist.`).to.be.true
expect(await readFile(transcript.path, 'utf8')).to.equal(`you
expect(await transcript.read()).to.equals(`you
`)
})
it('May transcribe a media file in french', async function () {
this.timeout(45000)
const transcript = await transcriber.transcribe(frVideoPath, { name: 'tiny' }, 'fr', 'txt')
expect(transcript).to.deep.equals({
expect(transcript.equals(new TranscriptFile({
path: join(transcriptDirectory, 'communiquer-lors-dune-classe-transplantee.txt'),
language: 'fr',
format: 'txt'
})
}))).to.be.true
expect(existsSync(transcript.path), `Transcript file ${transcript.path} doesn't exist.`).to.be.true
expect(await readFile(transcript.path, 'utf8')).to.equal(
expect(await transcript.read()).to.equal(
`...
Communiquez lors du ne class et transplanté.
Utilisez les photos prises lors de cette classe pour raconter quotidiennement le seuil jour vécu.
@ -143,7 +135,7 @@ Ensuite, il pourront lire et commenter ce de leur camarade, ou répondre au comm
)
const openaiTranscript = await openaiTranscriber.transcribe(...transcribeParameters)
expect(await readFile(transcript.path, 'utf8')).to.equal(await readFile(openaiTranscript.path, 'utf8'))
expect(await transcript.read()).to.equals(await openaiTranscript.read())
})
after(async function () {