chore: import upstream snapshot with attribution
Update Schema / Update configuration json schema (push) Has been cancelled
Memory Benchmark / Memory Test (Full Analysis) (push) Has been cancelled
CI Quality / Lint GitHub Actions (actionlint) (push) Has been cancelled
CI Quality / Lint GitHub Actions (zizmor) (push) Has been cancelled
CI Quality / Check typos (push) Has been cancelled
CI / Test coverage (push) Has been cancelled
CI / Test (22.x, windows-latest) (push) Has been cancelled
CI / Test (24.x, macos-latest) (push) Has been cancelled
CI / Test (24.x, ubuntu-latest) (push) Has been cancelled
CI / Test (24.x, windows-latest) (push) Has been cancelled
CI / Test (26.x, macos-latest) (push) Has been cancelled
CI / Test (26.x, ubuntu-latest) (push) Has been cancelled
CI / Test (26.x, windows-latest) (push) Has been cancelled
CI / Test with Bun (latest, macos-latest) (push) Has been cancelled
CI / Test with Bun (latest, ubuntu-latest) (push) Has been cancelled
CI / Test with Bun (latest, windows-latest) (push) Has been cancelled
CI / Build and run (22.x, macos-latest) (push) Has been cancelled
CI / Build and run (22.x, ubuntu-latest) (push) Has been cancelled
CI / Build and run (22.x, windows-latest) (push) Has been cancelled
autofix.ci / autofix (push) Has been cancelled
CI Browser Extension / Lint Browser Extension (push) Has been cancelled
CI Browser Extension / Test Browser Extension (push) Has been cancelled
Memory Benchmark / Memory Test (push) Has been cancelled
CI Website / Lint Website Client (push) Has been cancelled
CI Website / Lint Website Server (push) Has been cancelled
CI Website / Test Website Server (push) Has been cancelled
CI Website / Bundle Website Server (push) Has been cancelled
CI / Lint Biome (push) Has been cancelled
CI / Lint oxlint (push) Has been cancelled
CI / Lint TypeScript (push) Has been cancelled
CI / Lint Secretlint (push) Has been cancelled
CI / Test (22.x, macos-latest) (push) Has been cancelled
CI / Test (22.x, ubuntu-latest) (push) Has been cancelled
CI / Build and run (24.x, macos-latest) (push) Has been cancelled
CI / Build and run (24.x, ubuntu-latest) (push) Has been cancelled
CI / Build and run (24.x, windows-latest) (push) Has been cancelled
CI / Build and run (26.x, macos-latest) (push) Has been cancelled
CI / Build and run (26.x, ubuntu-latest) (push) Has been cancelled
CI / Build and run (26.x, windows-latest) (push) Has been cancelled
CI / Build and run with Bun (latest, macos-latest) (push) Has been cancelled
CI / Build and run with Bun (latest, ubuntu-latest) (push) Has been cancelled
CI / Build and run with Bun (latest, windows-latest) (push) Has been cancelled
CodeQL / Analyze (javascript-typescript) (push) Has been cancelled
Docker / build (linux/amd64, ubuntu-latest) (push) Has been cancelled
Docker / build (linux/arm/v7, ubuntu-24.04-arm) (push) Has been cancelled
Docker / build (linux/arm64, ubuntu-24.04-arm) (push) Has been cancelled
Docker / merge (push) Has been cancelled
Pack repository with Repomix / pack-repo (push) Has been cancelled
Performance Benchmark History / Benchmark (macos-latest) (push) Has been cancelled
Performance Benchmark History / Benchmark (ubuntu-latest) (push) Has been cancelled
Performance Benchmark History / Benchmark (windows-latest) (push) Has been cancelled
Performance Benchmark History / Store Results (push) Has been cancelled
Test Repomix Action / Test Node.js 22 (push) Has been cancelled
Test Repomix Action / Test Node.js 24 (push) Has been cancelled
Test Repomix Action / Test Node.js 26 (push) Has been cancelled
Update Schema / Update configuration json schema (push) Has been cancelled
Memory Benchmark / Memory Test (Full Analysis) (push) Has been cancelled
CI Quality / Lint GitHub Actions (actionlint) (push) Has been cancelled
CI Quality / Lint GitHub Actions (zizmor) (push) Has been cancelled
CI Quality / Check typos (push) Has been cancelled
CI / Test coverage (push) Has been cancelled
CI / Test (22.x, windows-latest) (push) Has been cancelled
CI / Test (24.x, macos-latest) (push) Has been cancelled
CI / Test (24.x, ubuntu-latest) (push) Has been cancelled
CI / Test (24.x, windows-latest) (push) Has been cancelled
CI / Test (26.x, macos-latest) (push) Has been cancelled
CI / Test (26.x, ubuntu-latest) (push) Has been cancelled
CI / Test (26.x, windows-latest) (push) Has been cancelled
CI / Test with Bun (latest, macos-latest) (push) Has been cancelled
CI / Test with Bun (latest, ubuntu-latest) (push) Has been cancelled
CI / Test with Bun (latest, windows-latest) (push) Has been cancelled
CI / Build and run (22.x, macos-latest) (push) Has been cancelled
CI / Build and run (22.x, ubuntu-latest) (push) Has been cancelled
CI / Build and run (22.x, windows-latest) (push) Has been cancelled
autofix.ci / autofix (push) Has been cancelled
CI Browser Extension / Lint Browser Extension (push) Has been cancelled
CI Browser Extension / Test Browser Extension (push) Has been cancelled
Memory Benchmark / Memory Test (push) Has been cancelled
CI Website / Lint Website Client (push) Has been cancelled
CI Website / Lint Website Server (push) Has been cancelled
CI Website / Test Website Server (push) Has been cancelled
CI Website / Bundle Website Server (push) Has been cancelled
CI / Lint Biome (push) Has been cancelled
CI / Lint oxlint (push) Has been cancelled
CI / Lint TypeScript (push) Has been cancelled
CI / Lint Secretlint (push) Has been cancelled
CI / Test (22.x, macos-latest) (push) Has been cancelled
CI / Test (22.x, ubuntu-latest) (push) Has been cancelled
CI / Build and run (24.x, macos-latest) (push) Has been cancelled
CI / Build and run (24.x, ubuntu-latest) (push) Has been cancelled
CI / Build and run (24.x, windows-latest) (push) Has been cancelled
CI / Build and run (26.x, macos-latest) (push) Has been cancelled
CI / Build and run (26.x, ubuntu-latest) (push) Has been cancelled
CI / Build and run (26.x, windows-latest) (push) Has been cancelled
CI / Build and run with Bun (latest, macos-latest) (push) Has been cancelled
CI / Build and run with Bun (latest, ubuntu-latest) (push) Has been cancelled
CI / Build and run with Bun (latest, windows-latest) (push) Has been cancelled
CodeQL / Analyze (javascript-typescript) (push) Has been cancelled
Docker / build (linux/amd64, ubuntu-latest) (push) Has been cancelled
Docker / build (linux/arm/v7, ubuntu-24.04-arm) (push) Has been cancelled
Docker / build (linux/arm64, ubuntu-24.04-arm) (push) Has been cancelled
Docker / merge (push) Has been cancelled
Pack repository with Repomix / pack-repo (push) Has been cancelled
Performance Benchmark History / Benchmark (macos-latest) (push) Has been cancelled
Performance Benchmark History / Benchmark (ubuntu-latest) (push) Has been cancelled
Performance Benchmark History / Benchmark (windows-latest) (push) Has been cancelled
Performance Benchmark History / Store Results (push) Has been cancelled
Test Repomix Action / Test Node.js 22 (push) Has been cancelled
Test Repomix Action / Test Node.js 24 (push) Has been cancelled
Test Repomix Action / Test Node.js 26 (push) Has been cancelled
This commit is contained in:
@@ -0,0 +1,118 @@
|
||||
import { describe, expect, test } from 'vitest';
|
||||
import { parseFile } from '../../../src/core/treeSitter/parseFile.js';
|
||||
import { createMockConfig } from '../../testing/testUtils.js';
|
||||
|
||||
describe('parseFile for Python', () => {
|
||||
const defaultConfig = createMockConfig({});
|
||||
|
||||
test('should parse basic Python correctly', async () => {
|
||||
const fileContent = `
|
||||
# Python greeting function
|
||||
'''
|
||||
A simple greeting function that prints a hello message
|
||||
Args:
|
||||
name: The name to greet
|
||||
'''
|
||||
def greet(name): print(f"Hello, {name}")
|
||||
`;
|
||||
const filePath = 'dummy.py';
|
||||
const config = {};
|
||||
const result = await parseFile(fileContent, filePath, createMockConfig(config));
|
||||
expect(typeof result).toBe('string');
|
||||
|
||||
const expectContents = [
|
||||
'greet',
|
||||
'Python greeting function',
|
||||
'A simple greeting function',
|
||||
'Args:',
|
||||
'name: The name to greet',
|
||||
];
|
||||
|
||||
for (const expectContent of expectContents) {
|
||||
expect(result).toContain(expectContent);
|
||||
}
|
||||
});
|
||||
|
||||
test('should parse class with decorators and inheritance', async () => {
|
||||
const fileContent = `
|
||||
@dataclass
|
||||
@register
|
||||
class UserModel(BaseModel):
|
||||
name: str
|
||||
age: int
|
||||
`;
|
||||
|
||||
const result = await parseFile(fileContent, 'example.py', defaultConfig);
|
||||
|
||||
const expectContents = ['@dataclass', '@register', 'class UserModel(BaseModel)'];
|
||||
|
||||
for (const expectContent of expectContents) {
|
||||
expect(result).toContain(expectContent);
|
||||
}
|
||||
});
|
||||
|
||||
test('should parse function with decorators and type annotations', async () => {
|
||||
const fileContent = `
|
||||
@route("/users")
|
||||
@authenticate
|
||||
def get_users(page: int = 1, limit: int = 10) -> List[User]:
|
||||
return users[page:limit]
|
||||
`;
|
||||
|
||||
const result = await parseFile(fileContent, 'example.py', defaultConfig);
|
||||
|
||||
const expectContents = [
|
||||
'@route("/users")',
|
||||
'@authenticate',
|
||||
'def get_users(page: int = 1, limit: int = 10) -> List[User]',
|
||||
];
|
||||
|
||||
for (const expectContent of expectContents) {
|
||||
expect(result).toContain(expectContent);
|
||||
}
|
||||
});
|
||||
|
||||
test('should parse docstring', async () => {
|
||||
const fileContent = `
|
||||
"""
|
||||
This is a docstring
|
||||
with multiple lines
|
||||
"""
|
||||
`;
|
||||
|
||||
const result = await parseFile(fileContent, 'example.py', defaultConfig);
|
||||
|
||||
const expectContents = ['This is a docstring', 'with multiple lines'];
|
||||
|
||||
for (const expectContent of expectContents) {
|
||||
expect(result).toContain(expectContent);
|
||||
}
|
||||
});
|
||||
|
||||
test('should parse comments', async () => {
|
||||
const fileContent = '# This is a single line comment';
|
||||
|
||||
const result = await parseFile(fileContent, 'example.py', defaultConfig);
|
||||
|
||||
const expectContents = ['# This is a single line comment'];
|
||||
|
||||
for (const expectContent of expectContents) {
|
||||
expect(result).toContain(expectContent);
|
||||
}
|
||||
});
|
||||
|
||||
test('should parse type aliases', async () => {
|
||||
const fileContent = `
|
||||
UserId = int
|
||||
Result = Union[Success, Error]
|
||||
`;
|
||||
|
||||
const result = await parseFile(fileContent, 'example.py', defaultConfig);
|
||||
|
||||
const expectContents = ['UserId = int', 'Result = Union[Success, Error]'];
|
||||
|
||||
for (const expectContent of expectContents) {
|
||||
expect(result).toContain(expectContent);
|
||||
}
|
||||
});
|
||||
});
|
||||
Reference in New Issue
Block a user