File size: 506 Bytes
a50881d
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
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()