From 9b5fa6d35b2c26b5052a1fceb25b8de8a02024b4 Mon Sep 17 00:00:00 2001 From: Wessel Bruinsma Date: Tue, 14 Dec 2021 11:24:19 +0000 Subject: [PATCH] Fix test --- tests/test_types.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_types.py b/tests/test_types.py index b668887..37d9db6 100644 --- a/tests/test_types.py +++ b/tests/test_types.py @@ -193,8 +193,8 @@ def test_dtype_int(check_lazy_shapes): assert B.dtype_int(1) is int # Test conversion back to right framework type. This conversion is thoroughly # tested for `B.promote_dtypes`. - assert B.dtype_float(tf.constant(1.0, dtype=tf.float32)) is tf.int32 - assert B.dtype_float(tf.constant(1.0, dtype=tf.float64)) is tf.int64 + assert B.dtype_int(tf.constant(1.0, dtype=tf.float32)) is tf.int32 + assert B.dtype_int(tf.constant(1.0, dtype=tf.float64)) is tf.int64 @pytest.mark.parametrize(