Commitd5622a4

committed

ENH: Add broadcasting test

1 parent5e8301c commitd5622a4Copy full SHA for d5622a4

File tree

+14

-3

lines changed

+14

-3

lines changed

Lines changed: 14 additions & 3 deletions

Original file line number	Diff line number	Diff line change
`@@ -484,11 +484,22 @@ def check_einsum_sums(self, dtype, do_opt=False):`
`484`	`484`	`# singleton dimensions broadcast (gh-10343)`
`485`	`485`	`p=np.ones((10,2))`
`486`	`486`	`q=np.ones((1,2))`
`487`		`-assert_array_equal(np.einsum('ti,ti->i',p,q,optimize=True),`
`488`		`-np.einsum('ti,ti->i',p,q,optimize=False))`
`489`		`-assert_array_equal(np.einsum('ti,ti->i',p,q,optimize=True),`
	`487`	`+assert_array_equal(np.einsum('ij,ij->j',p,q,optimize=True),`
	`488`	`+np.einsum('ij,ij->j',p,q,optimize=False))`
	`489`	`+assert_array_equal(np.einsum('ij,ij->j',p,q,optimize=True),`
`490`	`490`	`[10.]*2)`
`491`	`491`
	`492`	`+p=np.ones((1,5))`
	`493`	`+q=np.ones((5,5))`
	`494`	`+foroptimizein (True,False):`
	`495`	`+assert_array_equal(np.einsum("...ij,...jk->...ik",p,p,`
	`496`	`+optimize=optimize),`
	`497`	`+np.einsum("...ij,...jk->...ik",p,q,`
	`498`	`+optimize=optimize))`
	`499`	`+assert_array_equal(np.einsum("...ij,...jk->...ik",p,q,`
	`500`	`+optimize=optimize),`
	`501`	`+np.full((1,5),5))`
	`502`	`+`
`492`	`503`	`deftest_einsum_sums_int8(self):`
`493`	`504`	`self.check_einsum_sums('i1')`
`494`	`505`

Comments

(0)