diff --git a/tests/cases/checker/09_frame_ops.py b/tests/cases/checker/09_frame_ops.py index 6d478f6..908f4ba 100644 --- a/tests/cases/checker/09_frame_ops.py +++ b/tests/cases/checker/09_frame_ops.py @@ -1,8 +1,8 @@ # type: ignore # ruff: disable [F821] -df1: Frame[a:int, b:float] -df2: Frame[a:int, b:float] +df1: Frame[i:int, a:int, b:float] +df2: Frame[i:int, a:int, b:float] _: Any @@ -38,7 +38,7 @@ _ = df1.sum() _ = df1.var() # Groupby -df_gb = df1.groupby(by="a") +df_gb = df1.groupby(by="i") _ = df_gb.kurt() _ = df_gb.max() diff --git a/tests/cases/checker/09_frame_ops.py.ref.json b/tests/cases/checker/09_frame_ops.py.ref.json index dd689df..189640b 100644 --- a/tests/cases/checker/09_frame_ops.py.ref.json +++ b/tests/cases/checker/09_frame_ops.py.ref.json @@ -14,7 +14,7 @@ "columns": [ { "index": 0, - "name": "a", + "name": "i", "type": { "type": { "name": "int" @@ -23,6 +23,15 @@ }, { "index": 1, + "name": "a", + "type": { + "type": { + "name": "int" + } + } + }, + { + "index": 2, "name": "b", "type": { "type": { @@ -46,7 +55,7 @@ "columns": [ { "index": 0, - "name": "a", + "name": "i", "type": { "type": { "name": "int" @@ -55,6 +64,15 @@ }, { "index": 1, + "name": "a", + "type": { + "type": { + "name": "int" + } + } + }, + { + "index": 2, "name": "b", "type": { "type": { @@ -86,7 +104,7 @@ "columns": [ { "index": 0, - "name": "a", + "name": "i", "type": { "type": { "name": "int" @@ -95,6 +113,15 @@ }, { "index": 1, + "name": "a", + "type": { + "type": { + "name": "int" + } + } + }, + { + "index": 2, "name": "b", "type": { "type": { @@ -118,7 +145,7 @@ "columns": [ { "index": 0, - "name": "a", + "name": "i", "type": { "type": { "name": "int" @@ -127,6 +154,15 @@ }, { "index": 1, + "name": "a", + "type": { + "type": { + "name": "int" + } + } + }, + { + "index": 2, "name": "b", "type": { "type": { @@ -150,7 +186,7 @@ "columns": [ { "index": 0, - "name": "a", + "name": "i", "type": { "type": { "name": "int" @@ -159,6 +195,15 @@ }, { "index": 1, + "name": "a", + "type": { + "type": { + "name": "int" + } + } + }, + { + "index": 2, "name": "b", "type": { "type": { @@ -190,7 +235,7 @@ "columns": [ { "index": 0, - "name": "a", + "name": "i", "type": { "type": { "name": "int" @@ -199,6 +244,15 @@ }, { "index": 1, + "name": "a", + "type": { + "type": { + "name": "int" + } + } + }, + { + "index": 2, "name": "b", "type": { "type": { @@ -222,7 +276,7 @@ "columns": [ { "index": 0, - "name": "a", + "name": "i", "type": { "type": { "name": "int" @@ -231,6 +285,15 @@ }, { "index": 1, + "name": "a", + "type": { + "type": { + "name": "int" + } + } + }, + { + "index": 2, "name": "b", "type": { "type": { @@ -254,7 +317,7 @@ "columns": [ { "index": 0, - "name": "a", + "name": "i", "type": { "type": { "name": "int" @@ -263,6 +326,15 @@ }, { "index": 1, + "name": "a", + "type": { + "type": { + "name": "int" + } + } + }, + { + "index": 2, "name": "b", "type": { "type": { @@ -294,7 +366,7 @@ "columns": [ { "index": 0, - "name": "a", + "name": "i", "type": { "type": { "name": "int" @@ -303,6 +375,15 @@ }, { "index": 1, + "name": "a", + "type": { + "type": { + "name": "int" + } + } + }, + { + "index": 2, "name": "b", "type": { "type": { @@ -326,7 +407,7 @@ "columns": [ { "index": 0, - "name": "a", + "name": "i", "type": { "type": { "name": "int" @@ -335,6 +416,15 @@ }, { "index": 1, + "name": "a", + "type": { + "type": { + "name": "int" + } + } + }, + { + "index": 2, "name": "b", "type": { "type": { @@ -358,7 +448,7 @@ "columns": [ { "index": 0, - "name": "a", + "name": "i", "type": { "type": { "name": "int" @@ -367,6 +457,15 @@ }, { "index": 1, + "name": "a", + "type": { + "type": { + "name": "int" + } + } + }, + { + "index": 2, "name": "b", "type": { "type": { @@ -398,7 +497,7 @@ "columns": [ { "index": 0, - "name": "a", + "name": "i", "type": { "type": { "name": "float" @@ -407,6 +506,15 @@ }, { "index": 1, + "name": "a", + "type": { + "type": { + "name": "float" + } + } + }, + { + "index": 2, "name": "b", "type": { "type": { @@ -430,7 +538,7 @@ "columns": [ { "index": 0, - "name": "a", + "name": "i", "type": { "type": { "name": "int" @@ -439,6 +547,15 @@ }, { "index": 1, + "name": "a", + "type": { + "type": { + "name": "int" + } + } + }, + { + "index": 2, "name": "b", "type": { "type": { @@ -462,7 +579,7 @@ "columns": [ { "index": 0, - "name": "a", + "name": "i", "type": { "type": { "name": "int" @@ -471,6 +588,15 @@ }, { "index": 1, + "name": "a", + "type": { + "type": { + "name": "int" + } + } + }, + { + "index": 2, "name": "b", "type": { "type": { @@ -502,7 +628,7 @@ "columns": [ { "index": 0, - "name": "a", + "name": "i", "type": { "type": { "name": "int" @@ -511,6 +637,15 @@ }, { "index": 1, + "name": "a", + "type": { + "type": { + "name": "int" + } + } + }, + { + "index": 2, "name": "b", "type": { "type": { @@ -534,7 +669,7 @@ "columns": [ { "index": 0, - "name": "a", + "name": "i", "type": { "type": { "name": "int" @@ -543,6 +678,15 @@ }, { "index": 1, + "name": "a", + "type": { + "type": { + "name": "int" + } + } + }, + { + "index": 2, "name": "b", "type": { "type": { @@ -566,7 +710,7 @@ "columns": [ { "index": 0, - "name": "a", + "name": "i", "type": { "type": { "name": "int" @@ -575,6 +719,15 @@ }, { "index": 1, + "name": "a", + "type": { + "type": { + "name": "int" + } + } + }, + { + "index": 2, "name": "b", "type": { "type": { @@ -606,7 +759,7 @@ "columns": [ { "index": 0, - "name": "a", + "name": "i", "type": { "type": { "name": "int" @@ -615,6 +768,15 @@ }, { "index": 1, + "name": "a", + "type": { + "type": { + "name": "int" + } + } + }, + { + "index": 2, "name": "b", "type": { "type": { @@ -638,7 +800,7 @@ "columns": [ { "index": 0, - "name": "a", + "name": "i", "type": { "type": { "name": "int" @@ -647,6 +809,15 @@ }, { "index": 1, + "name": "a", + "type": { + "type": { + "name": "int" + } + } + }, + { + "index": 2, "name": "b", "type": { "type": { @@ -670,7 +841,7 @@ "columns": [ { "index": 0, - "name": "a", + "name": "i", "type": { "type": { "name": "int" @@ -679,6 +850,15 @@ }, { "index": 1, + "name": "a", + "type": { + "type": { + "name": "int" + } + } + }, + { + "index": 2, "name": "b", "type": { "type": { @@ -710,7 +890,7 @@ "columns": [ { "index": 0, - "name": "a", + "name": "i", "type": { "type": { "name": "int" @@ -719,6 +899,15 @@ }, { "index": 1, + "name": "a", + "type": { + "type": { + "name": "int" + } + } + }, + { + "index": 2, "name": "b", "type": { "type": {} @@ -740,7 +929,7 @@ "columns": [ { "index": 0, - "name": "a", + "name": "i", "type": { "type": { "name": "int" @@ -749,6 +938,15 @@ }, { "index": 1, + "name": "a", + "type": { + "type": { + "name": "int" + } + } + }, + { + "index": 2, "name": "b", "type": { "type": { @@ -772,7 +970,7 @@ "columns": [ { "index": 0, - "name": "a", + "name": "i", "type": { "type": { "name": "int" @@ -781,6 +979,15 @@ }, { "index": 1, + "name": "a", + "type": { + "type": { + "name": "int" + } + } + }, + { + "index": 2, "name": "b", "type": { "type": { @@ -812,7 +1019,7 @@ "columns": [ { "index": 0, - "name": "a", + "name": "i", "type": { "type": { "name": "bool" @@ -821,6 +1028,15 @@ }, { "index": 1, + "name": "a", + "type": { + "type": { + "name": "bool" + } + } + }, + { + "index": 2, "name": "b", "type": { "type": { @@ -844,7 +1060,7 @@ "columns": [ { "index": 0, - "name": "a", + "name": "i", "type": { "type": { "name": "int" @@ -853,6 +1069,15 @@ }, { "index": 1, + "name": "a", + "type": { + "type": { + "name": "int" + } + } + }, + { + "index": 2, "name": "b", "type": { "type": { @@ -876,7 +1101,7 @@ "columns": [ { "index": 0, - "name": "a", + "name": "i", "type": { "type": { "name": "int" @@ -885,6 +1110,15 @@ }, { "index": 1, + "name": "a", + "type": { + "type": { + "name": "int" + } + } + }, + { + "index": 2, "name": "b", "type": { "type": { @@ -916,7 +1150,7 @@ "columns": [ { "index": 0, - "name": "a", + "name": "i", "type": { "type": { "name": "bool" @@ -925,6 +1159,15 @@ }, { "index": 1, + "name": "a", + "type": { + "type": { + "name": "bool" + } + } + }, + { + "index": 2, "name": "b", "type": { "type": { @@ -948,7 +1191,7 @@ "columns": [ { "index": 0, - "name": "a", + "name": "i", "type": { "type": { "name": "int" @@ -957,6 +1200,15 @@ }, { "index": 1, + "name": "a", + "type": { + "type": { + "name": "int" + } + } + }, + { + "index": 2, "name": "b", "type": { "type": { @@ -980,7 +1232,7 @@ "columns": [ { "index": 0, - "name": "a", + "name": "i", "type": { "type": { "name": "int" @@ -989,6 +1241,15 @@ }, { "index": 1, + "name": "a", + "type": { + "type": { + "name": "int" + } + } + }, + { + "index": 2, "name": "b", "type": { "type": { @@ -1020,7 +1281,7 @@ "columns": [ { "index": 0, - "name": "a", + "name": "i", "type": { "type": { "name": "bool" @@ -1029,6 +1290,15 @@ }, { "index": 1, + "name": "a", + "type": { + "type": { + "name": "bool" + } + } + }, + { + "index": 2, "name": "b", "type": { "type": { @@ -1052,7 +1322,7 @@ "columns": [ { "index": 0, - "name": "a", + "name": "i", "type": { "type": { "name": "int" @@ -1061,6 +1331,15 @@ }, { "index": 1, + "name": "a", + "type": { + "type": { + "name": "int" + } + } + }, + { + "index": 2, "name": "b", "type": { "type": { @@ -1084,7 +1363,7 @@ "columns": [ { "index": 0, - "name": "a", + "name": "i", "type": { "type": { "name": "int" @@ -1093,6 +1372,15 @@ }, { "index": 1, + "name": "a", + "type": { + "type": { + "name": "int" + } + } + }, + { + "index": 2, "name": "b", "type": { "type": { @@ -1124,7 +1412,7 @@ "columns": [ { "index": 0, - "name": "a", + "name": "i", "type": { "type": { "name": "bool" @@ -1133,6 +1421,15 @@ }, { "index": 1, + "name": "a", + "type": { + "type": { + "name": "bool" + } + } + }, + { + "index": 2, "name": "b", "type": { "type": { @@ -1156,7 +1453,7 @@ "columns": [ { "index": 0, - "name": "a", + "name": "i", "type": { "type": { "name": "int" @@ -1165,6 +1462,15 @@ }, { "index": 1, + "name": "a", + "type": { + "type": { + "name": "int" + } + } + }, + { + "index": 2, "name": "b", "type": { "type": { @@ -1188,7 +1494,7 @@ "columns": [ { "index": 0, - "name": "a", + "name": "i", "type": { "type": { "name": "int" @@ -1197,6 +1503,15 @@ }, { "index": 1, + "name": "a", + "type": { + "type": { + "name": "int" + } + } + }, + { + "index": 2, "name": "b", "type": { "type": { @@ -1228,7 +1543,7 @@ "columns": [ { "index": 0, - "name": "a", + "name": "i", "type": { "type": { "name": "bool" @@ -1237,6 +1552,15 @@ }, { "index": 1, + "name": "a", + "type": { + "type": { + "name": "bool" + } + } + }, + { + "index": 2, "name": "b", "type": { "type": { @@ -1260,7 +1584,7 @@ "columns": [ { "index": 0, - "name": "a", + "name": "i", "type": { "type": { "name": "int" @@ -1269,6 +1593,15 @@ }, { "index": 1, + "name": "a", + "type": { + "type": { + "name": "int" + } + } + }, + { + "index": 2, "name": "b", "type": { "type": { @@ -1292,7 +1625,7 @@ "columns": [ { "index": 0, - "name": "a", + "name": "i", "type": { "type": { "name": "int" @@ -1301,6 +1634,15 @@ }, { "index": 1, + "name": "a", + "type": { + "type": { + "name": "int" + } + } + }, + { + "index": 2, "name": "b", "type": { "type": { @@ -1332,7 +1674,7 @@ "columns": [ { "index": 0, - "name": "a", + "name": "i", "type": { "type": { "name": "bool" @@ -1341,6 +1683,15 @@ }, { "index": 1, + "name": "a", + "type": { + "type": { + "name": "bool" + } + } + }, + { + "index": 2, "name": "b", "type": { "type": { @@ -1364,7 +1715,7 @@ "columns": [ { "index": 0, - "name": "a", + "name": "i", "type": { "type": { "name": "int" @@ -1373,6 +1724,15 @@ }, { "index": 1, + "name": "a", + "type": { + "type": { + "name": "int" + } + } + }, + { + "index": 2, "name": "b", "type": { "type": { @@ -1418,7 +1778,7 @@ "columns": [ { "index": 0, - "name": "a", + "name": "i", "type": { "type": { "name": "int" @@ -1427,6 +1787,15 @@ }, { "index": 1, + "name": "a", + "type": { + "type": { + "name": "int" + } + } + }, + { + "index": 2, "name": "b", "type": { "type": { @@ -1472,7 +1841,7 @@ "columns": [ { "index": 0, - "name": "a", + "name": "i", "type": { "type": { "name": "int" @@ -1481,6 +1850,15 @@ }, { "index": 1, + "name": "a", + "type": { + "type": { + "name": "int" + } + } + }, + { + "index": 2, "name": "b", "type": { "type": { @@ -1526,7 +1904,7 @@ "columns": [ { "index": 0, - "name": "a", + "name": "i", "type": { "type": { "name": "int" @@ -1535,6 +1913,15 @@ }, { "index": 1, + "name": "a", + "type": { + "type": { + "name": "int" + } + } + }, + { + "index": 2, "name": "b", "type": { "type": { @@ -1580,7 +1967,7 @@ "columns": [ { "index": 0, - "name": "a", + "name": "i", "type": { "type": { "name": "int" @@ -1589,6 +1976,15 @@ }, { "index": 1, + "name": "a", + "type": { + "type": { + "name": "int" + } + } + }, + { + "index": 2, "name": "b", "type": { "type": { @@ -1634,7 +2030,7 @@ "columns": [ { "index": 0, - "name": "a", + "name": "i", "type": { "type": { "name": "int" @@ -1643,6 +2039,15 @@ }, { "index": 1, + "name": "a", + "type": { + "type": { + "name": "int" + } + } + }, + { + "index": 2, "name": "b", "type": { "type": { @@ -1688,7 +2093,7 @@ "columns": [ { "index": 0, - "name": "a", + "name": "i", "type": { "type": { "name": "int" @@ -1697,6 +2102,15 @@ }, { "index": 1, + "name": "a", + "type": { + "type": { + "name": "int" + } + } + }, + { + "index": 2, "name": "b", "type": { "type": { @@ -1742,7 +2156,7 @@ "columns": [ { "index": 0, - "name": "a", + "name": "i", "type": { "type": { "name": "int" @@ -1751,6 +2165,15 @@ }, { "index": 1, + "name": "a", + "type": { + "type": { + "name": "int" + } + } + }, + { + "index": 2, "name": "b", "type": { "type": { @@ -1796,7 +2219,7 @@ "columns": [ { "index": 0, - "name": "a", + "name": "i", "type": { "type": { "name": "int" @@ -1805,6 +2228,15 @@ }, { "index": 1, + "name": "a", + "type": { + "type": { + "name": "int" + } + } + }, + { + "index": 2, "name": "b", "type": { "type": { @@ -1850,7 +2282,7 @@ "columns": [ { "index": 0, - "name": "a", + "name": "i", "type": { "type": { "name": "int" @@ -1859,6 +2291,15 @@ }, { "index": 1, + "name": "a", + "type": { + "type": { + "name": "int" + } + } + }, + { + "index": 2, "name": "b", "type": { "type": { @@ -1904,7 +2345,7 @@ "columns": [ { "index": 0, - "name": "a", + "name": "i", "type": { "type": { "name": "int" @@ -1913,6 +2354,15 @@ }, { "index": 1, + "name": "a", + "type": { + "type": { + "name": "int" + } + } + }, + { + "index": 2, "name": "b", "type": { "type": { @@ -1958,7 +2408,7 @@ "columns": [ { "index": 0, - "name": "a", + "name": "i", "type": { "type": { "name": "int" @@ -1967,6 +2417,15 @@ }, { "index": 1, + "name": "a", + "type": { + "type": { + "name": "int" + } + } + }, + { + "index": 2, "name": "b", "type": { "type": { @@ -2006,7 +2465,7 @@ }, "expr": { "_type": "LiteralExpr", - "value": "a" + "value": "i" }, "type": { "name": "str" @@ -2025,7 +2484,7 @@ "columns": [ { "index": 0, - "name": "a", + "name": "i", "type": { "type": { "name": "int" @@ -2034,6 +2493,15 @@ }, { "index": 1, + "name": "a", + "type": { + "type": { + "name": "int" + } + } + }, + { + "index": 2, "name": "b", "type": { "type": { @@ -2063,7 +2531,7 @@ "keywords": { "by": { "_type": "LiteralExpr", - "value": "a" + "value": "i" } } }, @@ -2768,7 +3236,7 @@ "columns": [ { "index": 0, - "name": "a", + "name": "i", "type": { "type": { "name": "int" @@ -2777,6 +3245,15 @@ }, { "index": 1, + "name": "a", + "type": { + "type": { + "name": "int" + } + } + }, + { + "index": 2, "name": "b", "type": { "type": { @@ -2822,7 +3299,7 @@ "columns": [ { "index": 0, - "name": "a", + "name": "i", "type": { "type": { "name": "int" @@ -2831,6 +3308,15 @@ }, { "index": 1, + "name": "a", + "type": { + "type": { + "name": "int" + } + } + }, + { + "index": 2, "name": "b", "type": { "type": { @@ -4447,7 +4933,7 @@ "columns": [ { "index": 0, - "name": "a", + "name": "i", "type": { "type": { "name": "int" @@ -4456,6 +4942,15 @@ }, { "index": 1, + "name": "a", + "type": { + "type": { + "name": "int" + } + } + }, + { + "index": 2, "name": "b", "type": { "type": { @@ -4496,7 +4991,7 @@ "columns": [ { "index": 0, - "name": "a", + "name": "i", "type": { "type": { "name": "int" @@ -4505,6 +5000,15 @@ }, { "index": 1, + "name": "a", + "type": { + "type": { + "name": "int" + } + } + }, + { + "index": 2, "name": "b", "type": { "type": { @@ -4545,7 +5049,7 @@ "columns": [ { "index": 0, - "name": "a", + "name": "i", "type": { "type": { "name": "int" @@ -4554,6 +5058,15 @@ }, { "index": 1, + "name": "a", + "type": { + "type": { + "name": "int" + } + } + }, + { + "index": 2, "name": "b", "type": { "type": { @@ -4735,7 +5248,7 @@ "columns": [ { "index": 0, - "name": "a", + "name": "i", "type": { "type": { "name": "int" @@ -4744,6 +5257,15 @@ }, { "index": 1, + "name": "a", + "type": { + "type": { + "name": "int" + } + } + }, + { + "index": 2, "name": "b", "type": { "type": { @@ -4776,7 +5298,7 @@ "columns": [ { "index": 0, - "name": "a", + "name": "i", "type": { "type": { "name": "int" @@ -4785,6 +5307,15 @@ }, { "index": 1, + "name": "a", + "type": { + "type": { + "name": "int" + } + } + }, + { + "index": 2, "name": "b", "type": { "type": { @@ -4808,7 +5339,7 @@ "columns": [ { "index": 0, - "name": "a", + "name": "i", "type": { "type": { "name": "int" @@ -4817,6 +5348,15 @@ }, { "index": 1, + "name": "a", + "type": { + "type": { + "name": "int" + } + } + }, + { + "index": 2, "name": "b", "type": { "type": { @@ -4849,7 +5389,7 @@ "columns": [ { "index": 0, - "name": "a", + "name": "i", "type": { "type": { "name": "int" @@ -4858,6 +5398,15 @@ }, { "index": 1, + "name": "a", + "type": { + "type": { + "name": "int" + } + } + }, + { + "index": 2, "name": "b", "type": { "type": {