Adding more ts specs

NaturalNode · Mar 22, 2024 · dd1fc2c · dd1fc2c
1 parent 106ec4d
commit dd1fc2c
Show file tree

Hide file tree

Showing 26 changed files with 483 additions and 99 deletions.
diff --git a/lib/natural/tfidf/index.d.ts b/lib/natural/tfidf/index.d.ts
@@ -25,7 +25,7 @@ THE SOFTWARE.
 
 import type { Tokenizer } from '../tokenizers'
 
-declare type TfIdfCallback = (i: number, measure: number, key?: string) => void
+declare type TfIdfCallback = (i: number, measure: number, key?: string | Record<string, any>) => void
 
 declare interface TfIdfTerm {
   term: string
@@ -42,7 +42,7 @@ export class TfIdf {
 
   constructor (deserialized?: Record<string, unknown>)
   idf (term: string, force?: boolean): number
-  addDocument (document: string | string[], key?: string, restoreCache?: boolean): void
+  addDocument (document: string | string[] | Record<string, string>, key?: Record<string, any> | any, restoreCache?: boolean): void
   addFileSync (path: string, encoding?: string, key?: string, restoreCache?: boolean): void
   tfidf (terms: string | string[], d: number): number
   tfidfs (terms: string | string[], callback?: TfIdfCallback): number[]

diff --git a/lib/natural/tfidf/tfidf.js b/lib/natural/tfidf/tfidf.js
@@ -203,7 +203,7 @@ class TfIdf {
   setTokenizer (t) {
     if (!_.isFunction(t.tokenize)) { throw new Error('Expected a valid Tokenizer') }
     tokenizer = t
-  }
+  } 
 
   // Define a stopwords other than the default
   setStopwords (customStopwords) {

diff --git a/lib/natural/trie/index.d.ts b/lib/natural/trie/index.d.ts
@@ -27,7 +27,7 @@ export class Trie {
   addString (string: string): boolean
   addStrings (list: string[]): void
   contains (string: string): boolean
-  findPrefix (search: string): string[]
+  findPrefix (search: string): Array<string | null>
   findMatchesOnPath (search: string): string[]
   keysWithPrefix (prefix: string): string[]
   getSize (): number

diff --git a/package.json b/package.json
@@ -77,7 +77,7 @@
     "clean": "rimraf *~ #* *classifier.json dist io_spec/tmp/*.json",
     "test": "cross-env NODE_PATH=. jasmine --random=false spec/*_spec.js",
     "test_io": "jasmine --random=false io_spec/*_spec.js",
-    "test_ts": "cross-env NODE_PATH=. jasmine --random=false dist/spec/*_spec.js",
+    "test_ts": "cross-env NODE_PATH=. jasmine --random=false dist/ts_spec/*_spec.js",
     "coverage": "nyc --reporter=lcov npm run test && nyc npm run test_io",
     "test_browser": "cross-env NODE_PATH=. node ./node_modules/gulp/bin/gulp.js",
     "lint": "eslint . --ext .ts"

diff --git a/spec/brill_pos_tagger_test.ts b/spec/brill_pos_tagger_test.ts
diff --git a/spec/tfidf_test.ts b/spec/tfidf_test.ts
diff --git a/spec/trie_test.ts b/spec/trie_test.ts
diff --git a/spec/SentenceAnalyzer_test.ts → ts_spec/SentenceAnalyzer_test.ts b/spec/SentenceAnalyzer_test.ts → ts_spec/SentenceAnalyzer_test.ts
diff --git a/spec/SentimentAnalyzer_test.ts → ts_spec/SentimentAnalyzer_test.ts b/spec/SentimentAnalyzer_test.ts → ts_spec/SentimentAnalyzer_test.ts
diff --git a/spec/bayes_classifier_spec.ts → ts_spec/bayes_classifier_spec.ts b/spec/bayes_classifier_spec.ts → ts_spec/bayes_classifier_spec.ts
diff --git a/spec/brill_pos_tagger_spec.ts → ts_spec/brill_pos_tagger_spec.ts b/spec/brill_pos_tagger_spec.ts → ts_spec/brill_pos_tagger_spec.ts
@@ -21,11 +21,11 @@ along with this program.  If not, see <http://www.gnu.org/licenses/>.
 import { WordTokenizer, Lexicon, RuleSet, BrillPOSTagger } from 'lib/natural'
 import type { Sentence, TagResults } from 'lib/natural'
 
-import * as englishSentences from './test_data/NYT-20150205-picassos-granddaughter-plans-to-sell-art-worrying-the-market.json'
+import * as englishSentences from '../spec/test_data/NYT-20150205-picassos-granddaughter-plans-to-sell-art-worrying-the-market.json'
 
-import jsonData from './test_data/NYT-20150205-picassos-granddaughter-plans_expected_tag_results.json'
+import jsonData from '../spec/test_data/NYT-20150205-picassos-granddaughter-plans_expected_tag_results.json'
 
-import * as dutchSentences from './test_data/Volkskrant-20150205-Knot-geldpers-aanzetten-is-paardenmiddel-voor-half-procent-inflatie.json'
+import * as dutchSentences from '../spec/test_data/Volkskrant-20150205-Knot-geldpers-aanzetten-is-paardenmiddel-voor-half-procent-inflatie.json'
 
 const englishTagResults: TagResults = jsonData as TagResults
 const DEBUG = false

diff --git a/spec/classifier_spec.ts → ts_spec/classifier_spec.ts b/spec/classifier_spec.ts → ts_spec/classifier_spec.ts
diff --git a/spec/distance_test.ts → ts_spec/distance_test.ts b/spec/distance_test.ts → ts_spec/distance_test.ts
diff --git a/spec/inflectors_test.ts → ts_spec/inflectors_test.ts b/spec/inflectors_test.ts → ts_spec/inflectors_test.ts
diff --git a/spec/logistic_regression_classifier_spec.ts → ...ec/logistic_regression_classifier_spec.ts b/spec/logistic_regression_classifier_spec.ts → ...ec/logistic_regression_classifier_spec.ts
diff --git a/spec/ngrams_test.ts → ts_spec/ngrams_test.ts b/spec/ngrams_test.ts → ts_spec/ngrams_test.ts
diff --git a/spec/phonetics_test.ts → ts_spec/phonetics_test.ts b/spec/phonetics_test.ts → ts_spec/phonetics_test.ts
diff --git a/spec/spellcheck_test.ts → ts_spec/spellcheck_test.ts b/spec/spellcheck_test.ts → ts_spec/spellcheck_test.ts
diff --git a/spec/stemmers_test.ts → ts_spec/stemmers_test.ts b/spec/stemmers_test.ts → ts_spec/stemmers_test.ts