
from __future__ import absolute_import

from traits.testing.unittest_tools import unittest

try:
    import numpy
except ImportError:
    numpy_available = False
else:
    numpy_available = True

from ..api import Array, Bool, HasTraits


class Foo(HasTraits):
    a = Array()
    event_fired = Bool(False)

    def _a_changed(self):
        self.event_fired = True


class ArrayTestCase(unittest.TestCase):
    """ Test cases for delegated traits. """

    @unittest.skipUnless(numpy_available, "test requires the NumPy package")
    def test_zero_to_one_element(self):
        """ Test that an event fires when an Array trait changes from zero to
        one element.
        """

        f = Foo()
        f.a = numpy.zeros((2,), float)
        f.event_fired = False

        # Change the array.
        f.a = numpy.concatenate((f.a, numpy.array([100])))

        # Confirm that the static trait handler was invoked.
        self.assertEqual(f.event_fired, True)

        return

#### EOF ######################################################################


from __future__ import absolute_import

from traits.testing.unittest_tools import unittest

try:
    import numpy
except ImportError:
    numpy_available = False
else:
    numpy_available = True

from ..api import Array, Bool, HasTraits


if numpy_available:
    # Use of `Array` requires NumPy to be installed.

    class Foo(HasTraits):
        a = Array()
        event_fired = Bool(False)

        def _a_changed(self):
            self.event_fired = True


class ArrayTestCase(unittest.TestCase):
    """ Test cases for delegated traits. """

    @unittest.skipUnless(numpy_available, "numpy not available")
    def test_zero_to_one_element(self):
        """ Test that an event fires when an Array trait changes from zero to
        one element.
        """

        f = Foo()
        f.a = numpy.zeros((2,), float)
        f.event_fired = False

        # Change the array.
        f.a = numpy.concatenate((f.a, numpy.array([100])))

        # Confirm that the static trait handler was invoked.
        self.assertEqual(f.event_fired, True)

        return

#### EOF ######################################################################

Make definition of conditional on NumPy being installed; update skip message to match that used elsewhere
Make definition of  conditional on NumPy being installed; update skip message to match that used elsewhere

  
  from __future__ import absolute_import
  
  from traits.testing.unittest_tools import unittest
  
  try:
      import numpy
  except ImportError:
      numpy_available = False
  else:
      numpy_available = True
  
  from ..api import Array, Bool, HasTraits
  
  
+ if numpy_available:
+     # Use of `Array` requires NumPy to be installed.
- class Foo(HasTraits):
-     a = Array()
-     event_fired = Bool(False)
  
+     class Foo(HasTraits):
+         a = Array()
+         event_fired = Bool(False)
+ 
-     def _a_changed(self):
+         def _a_changed(self):
-         self.event_fired = True
+             self.event_fired = True
  
  
  class ArrayTestCase(unittest.TestCase):
      """ Test cases for delegated traits. """
  
-     @unittest.skipUnless(numpy_available, "test requires the NumPy package")
+     @unittest.skipUnless(numpy_available, "numpy not available")
      def test_zero_to_one_element(self):
          """ Test that an event fires when an Array trait changes from zero to
          one element.
          """
  
          f = Foo()
          f.a = numpy.zeros((2,), float)
          f.event_fired = False
  
          # Change the array.
          f.a = numpy.concatenate((f.a, numpy.array([100])))
  
          # Confirm that the static trait handler was invoked.
          self.assertEqual(f.event_fired, True)
  
          return
  
  #### EOF ######################################################################
  
Make definition of conditional on NumPy being installed; update skip message to match that used elsewhere
## Code Before:

from __future__ import absolute_import

from traits.testing.unittest_tools import unittest

try:
    import numpy
except ImportError:
    numpy_available = False
else:
    numpy_available = True

from ..api import Array, Bool, HasTraits


class Foo(HasTraits):
    a = Array()
    event_fired = Bool(False)

    def _a_changed(self):
        self.event_fired = True


class ArrayTestCase(unittest.TestCase):
    """ Test cases for delegated traits. """

    @unittest.skipUnless(numpy_available, "test requires the NumPy package")
    def test_zero_to_one_element(self):
        """ Test that an event fires when an Array trait changes from zero to
        one element.
        """

        f = Foo()
        f.a = numpy.zeros((2,), float)
        f.event_fired = False

        # Change the array.
        f.a = numpy.concatenate((f.a, numpy.array([100])))

        # Confirm that the static trait handler was invoked.
        self.assertEqual(f.event_fired, True)

        return

#### EOF ######################################################################

## Instruction:
Make definition of conditional on NumPy being installed; update skip message to match that used elsewhere
## Code After:

from __future__ import absolute_import

from traits.testing.unittest_tools import unittest

try:
    import numpy
except ImportError:
    numpy_available = False
else:
    numpy_available = True

from ..api import Array, Bool, HasTraits


if numpy_available:
    # Use of `Array` requires NumPy to be installed.

    class Foo(HasTraits):
        a = Array()
        event_fired = Bool(False)

        def _a_changed(self):
            self.event_fired = True


class ArrayTestCase(unittest.TestCase):
    """ Test cases for delegated traits. """

    @unittest.skipUnless(numpy_available, "numpy not available")
    def test_zero_to_one_element(self):
        """ Test that an event fires when an Array trait changes from zero to
        one element.
        """

        f = Foo()
        f.a = numpy.zeros((2,), float)
        f.event_fired = False

        # Change the array.
        f.a = numpy.concatenate((f.a, numpy.array([100])))

        # Confirm that the static trait handler was invoked.
        self.assertEqual(f.event_fired, True)

        return

#### EOF ######################################################################
