def test_incomplete(self): self.assertTrue(broadcastable([1, 1, 1], [1, 2, 3])) self.assertTrue(broadcastable([2, 3], [1, 2, 3])) self.assertTrue(broadcastable([1, 3], [1, 2, 3])) self.assertTrue(broadcastable([1, 1], [1, 2, 3])) self.assertTrue(broadcastable([], [1, 2, 3])) self.assertTrue(broadcastable([1], [1, 2, 3]))
def test_reverse_incomplete(self): self.assertFalse(broadcastable([1, 2, 3], [1, 1, 1])) self.assertFalse(broadcastable([1, 2, 3], [2, 3])) self.assertFalse(broadcastable([1, 2, 3], [1, 3])) self.assertFalse(broadcastable([1, 2, 3], [1, 1])) self.assertFalse(broadcastable([1, 2, 3], [])) self.assertFalse(broadcastable([1, 2, 3], [1]))
def test_invalid(self): self.assertFalse(broadcastable([3, 2, 1], [1, 2, 3])) self.assertFalse(broadcastable([5], [6])) self.assertFalse(broadcastable([5], [1])) self.assertFalse(broadcastable([64], [1, 55, 56, 56]))
def test_matching(self): self.assertTrue(broadcastable([1, 2, 3], [1, 2, 3]))