Python assert_equal_shape_prefix Examples

Programming Language: Python

Namespace/Package Name: chex._src.asserts

Method/Function: assert_equal_shape_prefix

Examples at hotexamples.com: 2

Python assert_equal_shape_prefix - 2 examples found. These are the top rated real world Python examples of chex._src.asserts.assert_equal_shape_prefix extracted from open source projects. You can rate examples to help us improve the quality of examples.

Example #1

Show file

File: asserts_test.py Project: graingert/chex

 def test_equal_shape_prefix_should_fail(self, arrays):
   arrays = as_arrays(arrays)
   with self.assertRaisesRegex(AssertionError, 'different shape prefixes'):
     asserts.assert_equal_shape_prefix(arrays, prefix_len=1)

Example #2

Show file

File: asserts_test.py Project: graingert/chex

 def test_equal_shape_prefix_should_pass(self, arrays):
   arrays = as_arrays(arrays)
   asserts.assert_equal_shape_prefix(arrays, prefix_len=1)