126664f626f28f20e9f460c1cf5f6bf4643ce096,testing/test_data_object.py,TestDataHolderSimple,test_types,#TestDataHolderSimple#,26

Before Change

            self.m._needs_recompile = False

    def test_types(self):
        assert self.m.X.value.dtype == settings.np_float
        assert self.m.Y.value.dtype == settings.np_float
        assert self.m.Z.value.dtype == settings.np_float

    def test_same_shape(self):

After Change

        return m, rng

    def test_types(self):
        with self.test_context():
            m, _ = self.setup()
            self.assertEqual(m.X.dtype, settings.np_float)
            self.assertEqual(m.Y.dtype, settings.np_float)
            self.assertEqual(m.Z.dtype, settings.np_float)
            self.assertEqual(m.X.read_value().dtype, settings.np_float)
            self.assertEqual(m.Y.read_value().dtype, settings.np_float)
            self.assertEqual(m.Z.read_value().dtype, settings.np_float)
            m.compile()
            self.assertEqual(m.X.dtype, settings.np_float)
            self.assertEqual(m.Y.dtype, settings.np_float)
            self.assertEqual(m.Z.dtype, settings.np_float)
            self.assertEqual(m.X.read_value().dtype, settings.np_float)
            self.assertEqual(m.Y.read_value().dtype, settings.np_float)
            self.assertEqual(m.Z.read_value().dtype, settings.np_float)

    def test_same_shape(self):
        with self.test_context():
            m, rng = self.setup()
            m.compile()

Instances