@@ -41,7 +41,7 @@ def test_to_xarray_index_types(self, index_flat, df):
41
41
df .index .name = "foo"
42
42
df .columns .name = "bar"
43
43
result = df .to_xarray ()
44
- assert result .dims ["foo" ] == 3
44
+ assert result .sizes ["foo" ] == 3
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" ])
@@ -60,7 +60,7 @@ def test_to_xarray_empty(self, df):
60
60
61
61
df .index .name = "foo"
62
62
result = df [0 :0 ].to_xarray ()
63
- assert result .dims ["foo" ] == 0
63
+ assert result .sizes ["foo" ] == 0
64
64
assert isinstance (result , Dataset )
65
65
66
66
def test_to_xarray_with_multiindex (self , df ):
@@ -69,8 +69,8 @@ def test_to_xarray_with_multiindex(self, df):
69
69
# MultiIndex
70
70
df .index = MultiIndex .from_product ([["a" ], range (3 )], names = ["one" , "two" ])
71
71
result = df .to_xarray ()
72
- assert result .dims ["one" ] == 1
73
- assert result .dims ["two" ] == 3
72
+ assert result .sizes ["one" ] == 1
73
+ assert result .sizes ["two" ] == 3
74
74
assert len (result .coords ) == 2
75
75
assert len (result .data_vars ) == 8
76
76
tm .assert_almost_equal (list (result .coords .keys ()), ["one" , "two" ])
0 commit comments