Skip to content

Commit 8bb20b6

Browse files
committed
uncomment passing test
1 parent d0da954 commit 8bb20b6

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

larray/tests/test_la.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -966,9 +966,11 @@ def test_set(self):
966966
# self._assert_equal_raw(la, raw)
967967

968968
# d) broadcasting with a missing dimension
969-
# la = self.larray.copy()
970-
# la.set(la[ages1_5_9].sum(geo), age=ages1_5_9)
971-
# self._assert_equal_raw(la, raw)
969+
la = self.larray.copy()
970+
raw = self.array.copy()
971+
raw[[1, 5, 9]] = np.sum(raw[[1, 5, 9]], axis=1, keepdims=True)
972+
la.set(la[ages1_5_9].sum(geo), age=ages1_5_9)
973+
self._assert_equal_raw(la, raw)
972974

973975
# 2) using a string key
974976
la = self.larray.copy()

0 commit comments

Comments
 (0)