Spaces:
Running
Running
import unittest | |
from test_base import BaseChunkerTest | |
from Chunker import CodeChunker | |
class TestCodeChunkerRuby(BaseChunkerTest): | |
def setUp(self): | |
super().setUp() | |
self.code_chunker = CodeChunker(file_extension='rb') | |
def test_chunk_ruby_code(self): | |
rb_code = self.mock_codebase['example.rb'] | |
chunks, _ = self.run_chunker_test(rb_code) | |
self.assertGreater(len(chunks), 1) # Ensure the code is actually chunked | |
if __name__ == '__main__': | |
unittest.main() |