test(loaders): Fix text and json loaders tests

davidmigloz · Aug 12, 2023 · 3c69adf · 3c69adf
1 parent 4e911ea
commit 3c69adf
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 23 deletions.
diff --git a/packages/langchain/test/documents/loaders/json_test.dart b/packages/langchain/test/documents/loaders/json_test.dart
@@ -6,16 +6,6 @@ void main() {
     test('Test that a json file can be loaded', () async {
       const filePath = './test/documents/loaders/assets/example_1.json';
 
-      final expectedDoc = Document(
-        pageContent: 'Foo\nBar\nBaz\n',
-        metadata: {
-          'source': filePath,
-          'name': 'example_1.json',
-          'size': 32,
-          'lastModified': DateTime.parse('2023-07-13 17:49:35.000'),
-        },
-      );
-
       const loader = JsonLoader(
         filePath,
         jpSchema: r'$..text',
@@ -24,7 +14,14 @@ void main() {
       expect(
         loader.lazyLoad(),
         emitsInOrder([
-          expectedDoc,
+          (final Document doc) {
+            expect(doc.pageContent, 'Foo\nBar\nBaz\n');
+            expect(doc.metadata['source'], filePath);
+            expect(doc.metadata['name'], 'example_1.json');
+            expect(doc.metadata['size'], 32);
+            expect(doc.metadata['lastModified'], isA<DateTime>());
+            return true;
+          },
           emitsDone,
         ]),
       );

diff --git a/packages/langchain/test/documents/loaders/text_test.dart b/packages/langchain/test/documents/loaders/text_test.dart
@@ -5,22 +5,18 @@ void main() {
   group('TextLoader tests', () {
     test('Test that a text file can be loaded', () async {
       const filePath = './test/documents/loaders/assets/example.txt';
-
-      final expectedDoc = Document(
-        pageContent: 'Foo\nBar\nBaz\n',
-        metadata: {
-          'source': filePath,
-          'name': 'example.txt',
-          'size': 12,
-          'lastModified': DateTime.parse('2023-06-20 22:12:19.000'),
-        },
-      );
-
       const loader = TextLoader(filePath);
       expect(
         loader.lazyLoad(),
         emitsInOrder([
-          expectedDoc,
+          (final Document doc) {
+            expect(doc.pageContent, 'Foo\nBar\nBaz\n');
+            expect(doc.metadata['source'], filePath);
+            expect(doc.metadata['name'], 'example.txt');
+            expect(doc.metadata['size'], 12);
+            expect(doc.metadata['lastModified'], isA<DateTime>());
+            return true;
+          },
           emitsDone,
         ]),
       );