google / trax / 376

21%

Build Type

push

travis-ci

Committed by Copybara-Service

Commit Message

Overhaul of combinator tests.

Details:
  - Re-arrange the tests to align with their parter source file.
  - Rewrite tests to do direct computation rather than shape inference.
    (Part of ongoing pass to remove need for base.check_shape_agreement.)
  - Use "tl" alias for "trax.layers" so that the tests better resemble user
    code, where the "tl" alias is the norm.
  - Split out cases where one unit test was testing multiple separate behaviors.
  - Change Serial and Scan layers to allow lists of tensors as input, rather
    than just tuples; this aligns better with other layer types.
PiperOrigin-RevId: 310240056

Run Details

10 of 10 new or added lines in 1 file covered. (100.0%)

2680 of 10848 relevant lines covered (24.71%)

0.25 hits per line