@@ -41,7 +41,7 @@ def test_to_xarray_index_types(self, index_flat, df, using_infer_string):
41
41
df .index .name = "foo"
42
42
df .columns .name = "bar"
43
43
result = df .to_xarray ()
44
- assert result .dims ["foo" ] == 4
44
+ assert result .sizes ["foo" ] == 4
45
45
assert len (result .coords ) == 1
46
46
assert len (result .data_vars ) == 8
47
47
tm .assert_almost_equal (list (result .coords .keys ()), ["foo" ])
@@ -62,7 +62,7 @@ def test_to_xarray_empty(self, df):
62
62
63
63
df .index .name = "foo"
64
64
result = df [0 :0 ].to_xarray ()
65
- assert result .dims ["foo" ] == 0
65
+ assert result .sizes ["foo" ] == 0
66
66
assert isinstance (result , Dataset )
67
67
68
68
def test_to_xarray_with_multiindex (self , df , using_infer_string ):
@@ -71,8 +71,8 @@ def test_to_xarray_with_multiindex(self, df, using_infer_string):
71
71
# MultiIndex
72
72
df .index = MultiIndex .from_product ([["a" ], range (4 )], names = ["one" , "two" ])
73
73
result = df .to_xarray ()
74
- assert result .dims ["one" ] == 1
75
- assert result .dims ["two" ] == 4
74
+ assert result .sizes ["one" ] == 1
75
+ assert result .sizes ["two" ] == 4
76
76
assert len (result .coords ) == 2
77
77
assert len (result .data_vars ) == 8
78
78
tm .assert_almost_equal (list (result .coords .keys ()), ["one" , "two" ])
0 commit comments