CintraAI/code-chunker

1import unittest2from test_base import BaseChunkerTest3from Chunker import CodeChunker4 5class TestCodeChunkerRuby(BaseChunkerTest):6    def setUp(self):7        super().setUp()8        self.code_chunker = CodeChunker(file_extension='rb')9 10    def test_chunk_ruby_code(self):11        rb_code = self.mock_codebase['example.rb']12        chunks, _ = self.run_chunker_test(rb_code)13        self.assertGreater(len(chunks), 1)  # Ensure the code is actually chunked14 15if __name__ == '__main__':16    unittest.main()