(self)
| 78 | self.assertEqual(len(tasks), 40 - 24) |
| 79 | |
| 80 | async def test_file_writer(self): |
| 81 | writer = get_buffer_writer(self.config.buffer.trainer_input.experience_buffer) |
| 82 | await writer.acquire() |
| 83 | await writer.write( |
| 84 | [ |
| 85 | {"prompt": "hello world"}, |
| 86 | {"prompt": "hi"}, |
| 87 | ] |
| 88 | ) |
| 89 | await writer.write( |
| 90 | [ |
| 91 | {"prompt": "My name is"}, |
| 92 | {"prompt": "What is your name?"}, |
| 93 | ] |
| 94 | ) |
| 95 | await writer.release() |
| 96 | file_wrapper = ray.get_actor("json-test_buffer") |
| 97 | self.assertIsNotNone(file_wrapper) |
| 98 | file_path = self.config.buffer.trainer_input.experience_buffer.path |
| 99 | with open(file_path, "r") as f: |
| 100 | self.assertEqual(len(f.readlines()), 4) |
| 101 | |
| 102 | def setUp(self): |
| 103 | self.config = get_template_config() |
nothing calls this directly
no test coverage detected