| 133 | # 2 * sd2 |
| 134 | |
| 135 | def test_squeeze(self): |
| 136 | sd1 = idd.SingleData([1, 2, 3, 4], index=["foo", "bar", "f", "g"]) |
| 137 | # automatically squeezing |
| 138 | self.assertTrue(not isinstance(np.nansum(sd1), idd.IndexData)) |
| 139 | self.assertTrue(not isinstance(np.sum(sd1), idd.IndexData)) |
| 140 | self.assertTrue(not isinstance(sd1.sum(), idd.IndexData)) |
| 141 | self.assertEqual(np.nansum(sd1), 10) |
| 142 | self.assertEqual(np.sum(sd1), 10) |
| 143 | self.assertEqual(sd1.sum(), 10) |
| 144 | self.assertEqual(np.nanmean(sd1), 2.5) |
| 145 | self.assertEqual(np.mean(sd1), 2.5) |
| 146 | self.assertEqual(sd1.mean(), 2.5) |
| 147 | |
| 148 | |
| 149 | if __name__ == "__main__": |