(self)
| 242 | self.assert_integrity_idxs_take() |
| 243 | |
| 244 | def assert_integrity_idxs_take(self): |
| 245 | idxs_memo = self.idxs_memo |
| 246 | take_memo = self.take_memo |
| 247 | after = dfs(self.expr) |
| 248 | assert after == self.dfs_nodes |
| 249 | assert set(idxs_memo.keys()) == set(take_memo.keys()) |
| 250 | for node in idxs_memo: |
| 251 | idxs = idxs_memo[node] |
| 252 | assert idxs.name == "array_union" |
| 253 | vals = take_memo[node][0].pos_args[1] |
| 254 | for take in take_memo[node]: |
| 255 | assert take.name == "idxs_take" |
| 256 | assert [idxs, vals] == take.pos_args[:2] |
| 257 | |
| 258 | def build_idxs_vals(self, node, wanted_idxs): |
| 259 | """ |
no test coverage detected