@@ -4756,24 +4756,26 @@ def test_categorical(self):
4756
4756
tm .assert_series_equal (s , result )
4757
4757
4758
4758
_maybe_remove (store , 'df' )
4759
-
4760
4759
df = DataFrame ({"s" : s , "vals" : [1 , 2 , 3 , 4 , 5 , 6 ]})
4761
4760
store .append ('df' , df , format = 'table' )
4762
4761
result = store .select ('df' )
4763
4762
tm .assert_frame_equal (result , df )
4764
4763
4765
4764
# Dtypes
4765
+ _maybe_remove (store , 'si' )
4766
4766
s = Series ([1 , 1 , 2 , 2 , 3 , 4 , 5 ]).astype ('category' )
4767
4767
store .append ('si' , s )
4768
4768
result = store .select ('si' )
4769
4769
tm .assert_series_equal (result , s )
4770
4770
4771
+ _maybe_remove (store , 'si2' )
4771
4772
s = Series ([1 , 1 , np .nan , 2 , 3 , 4 , 5 ]).astype ('category' )
4772
4773
store .append ('si2' , s )
4773
4774
result = store .select ('si2' )
4774
4775
tm .assert_series_equal (result , s )
4775
4776
4776
4777
# Multiple
4778
+ _maybe_remove (store , 'df2' )
4777
4779
df2 = df .copy ()
4778
4780
df2 ['s2' ] = Series (list ('abcdefg' )).astype ('category' )
4779
4781
store .append ('df2' , df2 )
@@ -4787,13 +4789,15 @@ def test_categorical(self):
4787
4789
assert '/df2/meta/values_block_1/meta' in info
4788
4790
4789
4791
# unordered
4792
+ _maybe_remove (store , 's2' )
4790
4793
s = Series (Categorical (['a' , 'b' , 'b' , 'a' , 'a' , 'c' ], categories = [
4791
4794
'a' , 'b' , 'c' , 'd' ], ordered = False ))
4792
4795
store .append ('s2' , s , format = 'table' )
4793
4796
result = store .select ('s2' )
4794
4797
tm .assert_series_equal (result , s )
4795
4798
4796
4799
# Query
4800
+ _maybe_remove (store , 'df3' )
4797
4801
store .append ('df3' , df , data_columns = ['s' ])
4798
4802
expected = df [df .s .isin (['b' , 'c' ])]
4799
4803
result = store .select ('df3' , where = ['s in ["b","c"]' ])
0 commit comments