diff --git a/traittypes/tests/test_traittypes.py b/traittypes/tests/test_traittypes.py index fc6eb10..465e5cf 100644 --- a/traittypes/tests/test_traittypes.py +++ b/traittypes/tests/test_traittypes.py @@ -57,13 +57,11 @@ class Foo(HasTraits): b = Array(dtype='int') c = Array(None, allow_none=True) d = Array([]) - e = Array(Undefined) foo = Foo() self.assertTrue(np.array_equal(foo.a, np.array(0))) self.assertTrue(np.array_equal(foo.b, np.array(0))) self.assertTrue(foo.c is None) self.assertTrue(np.array_equal(foo.d, [])) - self.assertTrue(foo.e is Undefined) def test_allow_none(self): class Foo(HasTraits): @@ -128,12 +126,10 @@ class Foo(HasTraits): a = DataFrame() b = DataFrame(None, allow_none=True) c = DataFrame([]) - d = DataFrame(Undefined) foo = Foo() self.assertTrue(foo.a.equals(pd.DataFrame())) self.assertTrue(foo.b is None) self.assertTrue(foo.c.equals(pd.DataFrame([]))) - self.assertTrue(foo.d is Undefined) def test_allow_none(self): class Foo(HasTraits): @@ -165,12 +161,10 @@ class Foo(HasTraits): a = Series() b = Series(None, allow_none=True) c = Series([]) - d = Series(Undefined) foo = Foo() self.assertTrue(foo.a.equals(pd.Series())) self.assertTrue(foo.b is None) self.assertTrue(foo.c.equals(pd.Series([]))) - self.assertTrue(foo.d is Undefined) def test_allow_none(self): class Foo(HasTraits): @@ -201,11 +195,9 @@ def test_initial_values(self): class Foo(HasTraits): a = Dataset() b = Dataset(None, allow_none=True) - d = Dataset(Undefined) foo = Foo() self.assertTrue(foo.a.equals(xr.Dataset())) self.assertTrue(foo.b is None) - self.assertTrue(foo.d is Undefined) def test_allow_none(self): class Foo(HasTraits): @@ -236,8 +228,6 @@ def test_initial_values(self): class Foo(HasTraits): b = DataArray(None, allow_none=True) c = DataArray([]) - d = DataArray(Undefined) foo = Foo() self.assertTrue(foo.b is None) self.assertTrue(foo.c.equals(xr.DataArray([]))) - self.assertTrue(foo.d is Undefined)