Skip to content

Commit 0ce8aee

Browse files
bagel897bagel897
committed
fix tests
1 parent a3008bb commit 0ce8aee

File tree

5 files changed

+36
-35
lines changed

5 files changed

+36
-35
lines changed

‎src/graph_sitter/core/interfaces/editable.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1174,4 +1174,4 @@ def _get_ast_children(self) -> list[tuple[str | None, AST]]:
11741174
@final
11751175
def ast(self) -> AST:
11761176
children = self._get_ast_children()
1177-
return AST(codegen_sdk_type=self.__class__.__name__, span=self.span, tree_sitter_type=self.ts_node_type, children=children)
1177+
return AST(graph_sitter_type=self.__class__.__name__, span=self.span, tree_sitter_type=self.ts_node_type, children=children)

‎src/graph_sitter/output/ast.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88

99
class AST(BaseModel):
1010
model_config = ConfigDict(frozen=True)
11-
codegen_sdk_type: str
11+
graph_sitter_type: str
1212
span: Span
1313
tree_sitter_type: str
1414
children: list[tuple[str | None, Self]]

‎tests/unit/sdk/output/snapshots/test_ast/test_ast_basic/ast.json‎

Lines changed: 14 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
{
2-
"codegen_sdk_type": "PyClass",
2+
"graph_sitter_type": "PyClass",
33
"span": {
44
"range": {
55
"start_byte": 1,
@@ -20,7 +20,7 @@
2020
[
2121
"get_name()",
2222
{
23-
"codegen_sdk_type": "Name",
23+
"graph_sitter_type": "Name",
2424
"span": {
2525
"range": {
2626
"start_byte": 7,
@@ -43,7 +43,7 @@
4343
[
4444
"methods",
4545
{
46-
"codegen_sdk_type": "PyCodeBlock",
46+
"graph_sitter_type": "PyCodeBlock",
4747
"span": {
4848
"range": {
4949
"start_byte": 16,
@@ -64,7 +64,7 @@
6464
[
6565
"statements",
6666
{
67-
"codegen_sdk_type": "MultiLineCollection",
67+
"graph_sitter_type": "MultiLineCollection",
6868
"span": {
6969
"range": {
7070
"start_byte": 16,
@@ -85,7 +85,7 @@
8585
[
8686
null,
8787
{
88-
"codegen_sdk_type": "SymbolStatement",
88+
"graph_sitter_type": "SymbolStatement",
8989
"span": {
9090
"range": {
9191
"start_byte": 16,
@@ -106,7 +106,7 @@
106106
[
107107
null,
108108
{
109-
"codegen_sdk_type": "Name",
109+
"graph_sitter_type": "Name",
110110
"span": {
111111
"range": {
112112
"start_byte": 20,
@@ -129,7 +129,7 @@
129129
[
130130
null,
131131
{
132-
"codegen_sdk_type": "Collection",
132+
"graph_sitter_type": "Collection",
133133
"span": {
134134
"range": {
135135
"start_byte": 23,
@@ -150,7 +150,7 @@
150150
[
151151
null,
152152
{
153-
"codegen_sdk_type": "PyParameter",
153+
"graph_sitter_type": "PyParameter",
154154
"span": {
155155
"range": {
156156
"start_byte": 24,
@@ -176,7 +176,7 @@
176176
[
177177
null,
178178
{
179-
"codegen_sdk_type": "PyCodeBlock",
179+
"graph_sitter_type": "PyCodeBlock",
180180
"span": {
181181
"range": {
182182
"start_byte": 38,
@@ -197,7 +197,7 @@
197197
[
198198
"statements",
199199
{
200-
"codegen_sdk_type": "MultiLineCollection",
200+
"graph_sitter_type": "MultiLineCollection",
201201
"span": {
202202
"range": {
203203
"start_byte": 38,
@@ -218,7 +218,7 @@
218218
[
219219
null,
220220
{
221-
"codegen_sdk_type": "ReturnStatement",
221+
"graph_sitter_type": "ReturnStatement",
222222
"span": {
223223
"range": {
224224
"start_byte": 38,
@@ -239,7 +239,7 @@
239239
[
240240
"value",
241241
{
242-
"codegen_sdk_type": "Name",
242+
"graph_sitter_type": "Name",
243243
"span": {
244244
"range": {
245245
"start_byte": 45,
@@ -280,7 +280,7 @@
280280
[
281281
null,
282282
{
283-
"codegen_sdk_type": "Name",
283+
"graph_sitter_type": "Name",
284284
"span": {
285285
"range": {
286286
"start_byte": 20,
@@ -303,7 +303,7 @@
303303
[
304304
null,
305305
{
306-
"codegen_sdk_type": "Name",
306+
"graph_sitter_type": "Name",
307307
"span": {
308308
"range": {
309309
"start_byte": 24,

‎tests/unit/sdk/output/snapshots/test_ast/test_ast_nested/ast-nested.json‎

Lines changed: 18 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
{
2-
"codegen_sdk_type": "PyClass",
2+
"graph_sitter_type": "PyClass",
33
"span": {
44
"range": {
55
"start_byte": 1,
@@ -20,7 +20,7 @@
2020
[
2121
"get_name()",
2222
{
23-
"codegen_sdk_type": "Name",
23+
"graph_sitter_type": "Name",
2424
"span": {
2525
"range": {
2626
"start_byte": 7,
@@ -43,7 +43,7 @@
4343
[
4444
"methods",
4545
{
46-
"codegen_sdk_type": "PyCodeBlock",
46+
"graph_sitter_type": "PyCodeBlock",
4747
"span": {
4848
"range": {
4949
"start_byte": 16,
@@ -64,7 +64,7 @@
6464
[
6565
"statements",
6666
{
67-
"codegen_sdk_type": "MultiLineCollection",
67+
"graph_sitter_type": "MultiLineCollection",
6868
"span": {
6969
"range": {
7070
"start_byte": 16,
@@ -85,7 +85,7 @@
8585
[
8686
null,
8787
{
88-
"codegen_sdk_type": "SymbolStatement",
88+
"graph_sitter_type": "SymbolStatement",
8989
"span": {
9090
"range": {
9191
"start_byte": 16,
@@ -106,7 +106,7 @@
106106
[
107107
null,
108108
{
109-
"codegen_sdk_type": "Name",
109+
"graph_sitter_type": "Name",
110110
"span": {
111111
"range": {
112112
"start_byte": 22,
@@ -129,7 +129,7 @@
129129
[
130130
null,
131131
{
132-
"codegen_sdk_type": "PyCodeBlock",
132+
"graph_sitter_type": "PyCodeBlock",
133133
"span": {
134134
"range": {
135135
"start_byte": 35,
@@ -150,7 +150,7 @@
150150
[
151151
"statements",
152152
{
153-
"codegen_sdk_type": "MultiLineCollection",
153+
"graph_sitter_type": "MultiLineCollection",
154154
"span": {
155155
"range": {
156156
"start_byte": 35,
@@ -171,7 +171,7 @@
171171
[
172172
null,
173173
{
174-
"codegen_sdk_type": "SymbolStatement",
174+
"graph_sitter_type": "SymbolStatement",
175175
"span": {
176176
"range": {
177177
"start_byte": 35,
@@ -192,7 +192,7 @@
192192
[
193193
null,
194194
{
195-
"codegen_sdk_type": "Name",
195+
"graph_sitter_type": "Name",
196196
"span": {
197197
"range": {
198198
"start_byte": 39,
@@ -215,7 +215,7 @@
215215
[
216216
null,
217217
{
218-
"codegen_sdk_type": "Collection",
218+
"graph_sitter_type": "Collection",
219219
"span": {
220220
"range": {
221221
"start_byte": 42,
@@ -236,7 +236,7 @@
236236
[
237237
null,
238238
{
239-
"codegen_sdk_type": "PyParameter",
239+
"graph_sitter_type": "PyParameter",
240240
"span": {
241241
"range": {
242242
"start_byte": 43,
@@ -262,7 +262,7 @@
262262
[
263263
null,
264264
{
265-
"codegen_sdk_type": "PyCodeBlock",
265+
"graph_sitter_type": "PyCodeBlock",
266266
"span": {
267267
"range": {
268268
"start_byte": 62,
@@ -283,7 +283,7 @@
283283
[
284284
"statements",
285285
{
286-
"codegen_sdk_type": "MultiLineCollection",
286+
"graph_sitter_type": "MultiLineCollection",
287287
"span": {
288288
"range": {
289289
"start_byte": 62,
@@ -304,7 +304,7 @@
304304
[
305305
null,
306306
{
307-
"codegen_sdk_type": "PyPassStatement",
307+
"graph_sitter_type": "PyPassStatement",
308308
"span": {
309309
"range": {
310310
"start_byte": 62,
@@ -351,7 +351,7 @@
351351
[
352352
null,
353353
{
354-
"codegen_sdk_type": "Name",
354+
"graph_sitter_type": "Name",
355355
"span": {
356356
"range": {
357357
"start_byte": 22,
@@ -374,7 +374,7 @@
374374
[
375375
null,
376376
{
377-
"codegen_sdk_type": "Name",
377+
"graph_sitter_type": "Name",
378378
"span": {
379379
"range": {
380380
"start_byte": 39,
@@ -397,7 +397,7 @@
397397
[
398398
null,
399399
{
400-
"codegen_sdk_type": "Name",
400+
"graph_sitter_type": "Name",
401401
"span": {
402402
"range": {
403403
"start_byte": 43,

‎tests/unit/sdk/python/detached_symbols/code_block/test_code_block.py‎

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -369,7 +369,8 @@ def my_db_query(
369369
) -> None:
370370
from_record = (
371371
parent.proxy
372-
) # added by Graph-sitter from_logs: list[Log] = (
372+
) # added by Graph-sitter
373+
from_logs: list[Log] = (
373374
db.session.execute(
374375
select(Log).where(
375376
Log.proxy_id == from_record.id

0 commit comments

Comments
 (0)