@@ -732,11 +732,11 @@ def test_unbalanced(self):
732
732
)
733
733
df ["id" ] = df .index
734
734
exp_data = {
735
- "X" : ["X1" , "X2 " , "X1 " , "X2" ],
736
- "A" : [1.0 , 2 .0 , 3 .0 , 4.0 ],
737
- "B" : [5.0 , 6.0 , np . nan , np .nan ],
738
- "id" : [0 , 1 , 0 , 1 ],
739
- "year" : [2010 , 2010 , 2011 , 2011 ],
735
+ "X" : ["X1" , "X1 " , "X2 " , "X2" ],
736
+ "A" : [1.0 , 3 .0 , 2 .0 , 4.0 ],
737
+ "B" : [5.0 , np . nan , 6.0 , np .nan ],
738
+ "id" : [0 , 0 , 1 , 1 ],
739
+ "year" : [2010 , 2011 , 2010 , 2011 ],
740
740
}
741
741
expected = pd .DataFrame (exp_data )
742
742
expected = expected .set_index (["id" , "year" ])[["X" , "A" , "B" ]]
@@ -979,10 +979,10 @@ def test_nonnumeric_suffix(self):
979
979
)
980
980
expected = pd .DataFrame (
981
981
{
982
- "A" : ["X1" , "X2 " , "X1 " , "X2" ],
983
- "colname" : ["placebo" , "placebo " , "test " , "test" ],
984
- "result" : [5.0 , 6.0 , np . nan , np .nan ],
985
- "treatment" : [1.0 , 2 .0 , 3 .0 , 4.0 ],
982
+ "A" : ["X1" , "X1 " , "X2 " , "X2" ],
983
+ "colname" : ["placebo" , "test " , "placebo " , "test" ],
984
+ "result" : [5.0 , np . nan , 6.0 , np .nan ],
985
+ "treatment" : [1.0 , 3 .0 , 2 .0 , 4.0 ],
986
986
}
987
987
)
988
988
expected = expected .set_index (["A" , "colname" ])
@@ -1026,10 +1026,10 @@ def test_float_suffix(self):
1026
1026
)
1027
1027
expected = pd .DataFrame (
1028
1028
{
1029
- "A" : ["X1" , "X2 " , "X1" , "X2 " , "X1 " , "X2" , "X1 " , "X2" ],
1030
- "colname" : [1.2 , 1.2 , 1.0 , 1.0 , 1.1 , 1.1 , 2.1 , 2.1 ],
1031
- "result" : [5.0 , 6 .0 , 0.0 , 9 .0 , np .nan , np . nan , np .nan , np .nan ],
1032
- "treatment" : [np .nan , np . nan , np .nan , np . nan , 1.0 , 2.0 , 3.0 , 4.0 ],
1029
+ "A" : ["X1" , "X1 " , "X1" , "X1 " , "X2 " , "X2" , "X2 " , "X2" ],
1030
+ "colname" : [1 , 1.1 , 1.2 , 2.1 , 1 , 1.1 , 1.2 , 2.1 ],
1031
+ "result" : [0 .0 , np . nan , 5 .0 , np .nan , 9.0 , np .nan , 6.0 , np .nan ],
1032
+ "treatment" : [np .nan , 1.0 , np .nan , 3.0 , np . nan , 2.0 , np . nan , 4.0 ],
1033
1033
}
1034
1034
)
1035
1035
expected = expected .set_index (["A" , "colname" ])
0 commit comments