16 Commits

Author SHA1 Message Date
Jamie Townsend
7506a3e5f0
Fix flaky generated_fun_test.py test (#3885) 2020-07-28 07:27:07 -07:00
Jake VanderPlas
afce718eb1 Add ability to specify individual test targets 2020-06-29 11:08:57 -07:00
Peter Hawkins
b543652332
Replace np -> jnp, onp -> np in tests. (#2969) 2020-05-05 14:59:16 -04:00
Peter Hawkins
e60d5dd54c
Remove "from __future__" uses from JAX. (#2117)
The future (Python 3) has arrived; no need to request it explicitly.
2020-01-29 12:29:03 -05:00
George Necula
c1d8d3f74d Add error checking that arguments of jvp are tuples 2019-11-27 13:12:24 +01:00
Matthew Johnson
d3ec0b23c5 fix miscellaneous bugs, incl. complex abs grad 2018-12-12 12:03:57 -05:00
Dougal Maclaurin
30124b6da1 Added jit transformations to generated functions. Fixed bug in comparing numpy arrays for equality. 2018-12-08 00:03:34 -05:00
Roy Frostig
e5d89ce908 update call to choice 2018-12-07 10:51:02 -08:00
Roy Frostig
6c2cdfc318 py3 compatibility: list of keys 2018-12-07 10:45:48 -08:00
Roy Frostig
fe0a26936c use custom choice function in generated tests 2018-12-07 10:40:21 -08:00
Roy Frostig
479f256abd py3 iterator compatibility 2018-12-07 10:22:15 -08:00
Roy Frostig
6fb8a8bc89 use print as function 2018-12-07 09:43:28 -08:00
Dougal Maclaurin
8b88027df0 Number of test cases settable with command-line flag 2018-12-06 18:30:59 -05:00
Dougal Maclaurin
ebc6cd1e03 Step through sizes in test case generation 2018-12-06 18:02:43 -05:00
Dougal Maclaurin
3dbf41f3e6 Generated function tests working with bazel 2018-12-06 17:31:52 -05:00
Dougal Maclaurin
c3374a9d5f Added build rule for generated_fun_test (formerly quickish_check) 2018-12-06 17:04:00 -05:00