mirror of
https://github.com/google/pebble.git
synced 2025-03-23 04:02:19 +00:00
229 lines
7.7 KiB
C
229 lines
7.7 KiB
C
/*
|
|
* Copyright 2024 Google LLC
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
#include "clar.h"
|
|
|
|
#include "services/normal/voice/transcription.h"
|
|
|
|
#include "test_transcription_example.h"
|
|
|
|
static int s_count;
|
|
|
|
// setup and teardown
|
|
void test_transcription__initialize(void) {
|
|
}
|
|
|
|
void test_transcription__cleanup(void) {
|
|
}
|
|
|
|
static bool prv_cb_return_true(void *w, void *data) {
|
|
return true;
|
|
}
|
|
|
|
static bool prv_cb_return_false(void *w, void *data) {
|
|
return (s_count++ != (int)data);
|
|
}
|
|
|
|
void test_transcription__validate(void) {
|
|
bool result;
|
|
Transcription * validate_test = (Transcription *) s_test_transcription_example;
|
|
size_t test_size = sizeof(s_test_transcription_example);
|
|
result = transcription_validate(validate_test, test_size);
|
|
cl_assert_equal_p(result, true);
|
|
|
|
result = transcription_validate(NULL, test_size);
|
|
cl_assert_equal_p(result, false);
|
|
|
|
result = transcription_validate(validate_test, test_size - 1);
|
|
cl_assert_equal_p(result, false);
|
|
|
|
result = transcription_validate(validate_test, test_size + 1);
|
|
cl_assert_equal_p(result, false);
|
|
|
|
result = transcription_validate(validate_test, 0);
|
|
cl_assert_equal_p(result, false);
|
|
|
|
result = transcription_validate(validate_test, sizeof(Transcription));
|
|
cl_assert_equal_p(result, false);
|
|
|
|
result = transcription_validate(validate_test, sizeof(Transcription) - 1);
|
|
cl_assert_equal_p(result, false);
|
|
|
|
s_test_transcription_example[0] = 0;
|
|
result = transcription_validate(validate_test, test_size);
|
|
cl_assert_equal_p(result, false);
|
|
s_test_transcription_example[0] = 1;
|
|
|
|
s_test_transcription_example[2] = 3; // invalidate word count of the first sentence
|
|
result = transcription_validate(validate_test, test_size);
|
|
cl_assert_equal_p(result, false);
|
|
s_test_transcription_example[2] = 2; // restore word count of the first sentence
|
|
|
|
s_test_transcription_example[5] = 4; // invalidate word length of first word in first sentence
|
|
result = transcription_validate(validate_test, test_size);
|
|
cl_assert_equal_p(result, false);
|
|
s_test_transcription_example[5] = 5; // restore word length of first word in first
|
|
|
|
s_test_transcription_example[23] = 2; // invalidate Word count of the second sentence
|
|
result = transcription_validate(validate_test, test_size);
|
|
cl_assert_equal_p(result, false);
|
|
s_test_transcription_example[23] = 3; // restore word count of the second sentence
|
|
|
|
s_test_transcription_example[26] = 0; // invalidate word length of first word in second sentence
|
|
result = transcription_validate(validate_test, test_size);
|
|
cl_assert_equal_p(result, false);
|
|
|
|
s_test_transcription_example[26] = 5; // invalidate word length of first word in second sentence
|
|
result = transcription_validate(validate_test, test_size);
|
|
cl_assert_equal_p(result, false);
|
|
s_test_transcription_example[26] = 4; // restore word length of first word in second sentence
|
|
|
|
s_test_transcription_example[8] = '\n'; // insert special character into the middle of a word
|
|
result = transcription_validate(validate_test, test_size);
|
|
cl_assert_equal_p(result, false);
|
|
|
|
s_test_transcription_example[8] = '\0'; // insert null terminator into the middle of a word
|
|
result = transcription_validate(validate_test, test_size);
|
|
cl_assert_equal_p(result, false);
|
|
s_test_transcription_example[8] = 'e'; // restore word
|
|
|
|
s_test_transcription_example[11] = '\0'; // insert null terminator at the end of a word
|
|
result = transcription_validate(validate_test, test_size);
|
|
cl_assert_equal_p(result, false);
|
|
|
|
s_test_transcription_example[9] = '\xe2'; // insert valid utf8 into a word
|
|
s_test_transcription_example[10] = '\x9d';
|
|
s_test_transcription_example[11] = '\xa4';
|
|
result = transcription_validate(validate_test, test_size);
|
|
cl_assert_equal_p(result, true);
|
|
|
|
s_test_transcription_example[test_size - 1] = '\0'; // insert null terminator at the end of last word
|
|
result = transcription_validate(validate_test, test_size);
|
|
cl_assert_equal_p(result, false);
|
|
}
|
|
|
|
// tests
|
|
void test_transcription__iterate_words(void) {
|
|
uint8_t words_test[] = {
|
|
0,
|
|
0x04, 0x00,
|
|
't', 'e', 's', 't',
|
|
|
|
51,
|
|
0x05, 0x00,
|
|
'h', 'e', 'l', 'l', 'o',
|
|
|
|
101,
|
|
0x03, 0x00,
|
|
't', 'h', 'e'
|
|
};
|
|
|
|
uint8_t *end = words_test + sizeof(words_test);
|
|
uint8_t *result;
|
|
result = transcription_iterate_words((TranscriptionWord *)words_test, 3,
|
|
(TranscriptionWordIterateCb)prv_cb_return_true, NULL);
|
|
cl_assert_equal_p(result, end);
|
|
|
|
result = transcription_iterate_words((TranscriptionWord *)words_test, 2,
|
|
(TranscriptionWordIterateCb)prv_cb_return_true, NULL);
|
|
cl_assert_equal_p(result, &words_test[15]);
|
|
|
|
s_count = 0;
|
|
result = transcription_iterate_words((TranscriptionWord *)words_test, 3,
|
|
(TranscriptionWordIterateCb)prv_cb_return_false, (void *) 0);
|
|
cl_assert_equal_p(result, words_test);
|
|
|
|
s_count = 0;
|
|
result = transcription_iterate_words((TranscriptionWord *)words_test, 3,
|
|
(TranscriptionWordIterateCb)prv_cb_return_false, (void *) 1);
|
|
cl_assert_equal_p(result, &words_test[7]);
|
|
|
|
s_count = 0;
|
|
result = transcription_iterate_words((TranscriptionWord *)words_test, 3,
|
|
(TranscriptionWordIterateCb)prv_cb_return_false, (void *) 2);
|
|
cl_assert_equal_p(result, &words_test[15]);
|
|
|
|
result = transcription_iterate_words((TranscriptionWord *)words_test, 3, NULL, NULL);
|
|
cl_assert_equal_p(result, end);
|
|
}
|
|
|
|
void test_transcription__iterate_sentences(void) {
|
|
uint8_t sentence_test[] = {
|
|
// Sentence #1
|
|
0x02, 0x00, // Word count
|
|
|
|
// Word #1
|
|
85, // Confidence
|
|
0x05, 0x00, // Word length
|
|
'H', 'e', 'l', 'l', 'o',
|
|
|
|
// Word #2
|
|
74, // Confidence
|
|
0x08, 0x00, // Word length
|
|
'c', 'o', 'm', 'p', 'u', 't', 'e', 'r',
|
|
|
|
// Sentence #2
|
|
0x03, 0x00, // Word count
|
|
|
|
// Word #1
|
|
13, // Confidence
|
|
0x04, 0x00, // Word length
|
|
'h', 'e', 'l', 'l',
|
|
|
|
// Word #1
|
|
3, // Confidence
|
|
0x02, 0x00, // Word length
|
|
'o', 'h',
|
|
|
|
// Word #2
|
|
0, // Confidence
|
|
0x07, 0x00, // Word length
|
|
'c', 'o', 'm', 'p', 'u', 't', 'a',
|
|
};
|
|
|
|
uint8_t *end = sentence_test + sizeof(sentence_test);
|
|
|
|
uint8_t *result;
|
|
result = transcription_iterate_sentences((TranscriptionSentence *) sentence_test, 2,
|
|
(TranscriptionSentenceIterateCb)prv_cb_return_true, NULL);
|
|
cl_assert_equal_p(result, end);
|
|
|
|
result = transcription_iterate_sentences((TranscriptionSentence *) sentence_test, 1,
|
|
(TranscriptionSentenceIterateCb)prv_cb_return_true, NULL);
|
|
cl_assert_equal_p(result, &sentence_test[21]);
|
|
|
|
s_count = 0;
|
|
result = transcription_iterate_sentences((TranscriptionSentence *) sentence_test, 2,
|
|
(TranscriptionSentenceIterateCb)prv_cb_return_false, (void *) 0);
|
|
cl_assert_equal_p(result, sentence_test);
|
|
|
|
s_count = 0;
|
|
result = transcription_iterate_sentences((TranscriptionSentence *) sentence_test, 2,
|
|
(TranscriptionSentenceIterateCb)prv_cb_return_false, (void *) 1);
|
|
cl_assert_equal_p(result, &sentence_test[21]);
|
|
|
|
result = transcription_iterate_sentences((TranscriptionSentence *) sentence_test, 0,
|
|
(TranscriptionSentenceIterateCb)prv_cb_return_false, NULL);
|
|
cl_assert_equal_p(result, sentence_test);
|
|
|
|
result = transcription_iterate_sentences((TranscriptionSentence *) sentence_test, 2,
|
|
NULL, NULL);
|
|
cl_assert_equal_p(result, end);
|
|
|
|
}
|
|
|
|
|
|
|