diff --git a/IPython/lib/tests/test_display.py b/IPython/lib/tests/test_display.py
index 7e98a18..414d3fd 100644
--- a/IPython/lib/tests/test_display.py
+++ b/IPython/lib/tests/test_display.py
@@ -59,8 +59,9 @@ def test_existing_path_FileLink():
tf = NamedTemporaryFile()
fl = display.FileLink(tf.name)
actual = fl._repr_html_()
- expected = "%s
" % (tf.name,tf.name)
- nt.assert_equal(actual,expected)
+ expected = "%s
" % (tf.name, tf.name)
+ assert actual == expected
+
def test_existing_path_FileLink_repr():
"""FileLink: Calling repr() functions as expected on existing filepath
@@ -69,7 +70,8 @@ def test_existing_path_FileLink_repr():
fl = display.FileLink(tf.name)
actual = repr(fl)
expected = tf.name
- nt.assert_equal(actual,expected)
+ assert actual == expected
+
def test_error_on_directory_to_FileLink():
"""FileLink: Raises error when passed directory
@@ -111,7 +113,8 @@ def test_existing_path_FileLinks():
(tf1.name.replace("\\","/"),split(tf1.name)[1])]
expected.sort()
# We compare the sorted list of links here as that's more reliable
- nt.assert_equal(actual,expected)
+ assert actual == expected
+
def test_existing_path_FileLinks_alt_formatter():
"""FileLinks: Calling _repr_html_ functions as expected w/ an alt formatter
@@ -128,7 +131,8 @@ def test_existing_path_FileLinks_alt_formatter():
expected = ["hello","world"]
expected.sort()
# We compare the sorted list of links here as that's more reliable
- nt.assert_equal(actual,expected)
+ assert actual == expected
+
def test_existing_path_FileLinks_repr():
"""FileLinks: Calling repr() functions as expected on existing directory """
@@ -142,8 +146,9 @@ def test_existing_path_FileLinks_repr():
expected = ['%s/' % td, ' %s' % split(tf1.name)[1],' %s' % split(tf2.name)[1]]
expected.sort()
# We compare the sorted list of links here as that's more reliable
- nt.assert_equal(actual,expected)
-
+ assert actual == expected
+
+
def test_existing_path_FileLinks_repr_alt_formatter():
"""FileLinks: Calling repr() functions as expected w/ alt formatter
"""
@@ -159,8 +164,9 @@ def test_existing_path_FileLinks_repr_alt_formatter():
expected = ["hello","world"]
expected.sort()
# We compare the sorted list of links here as that's more reliable
- nt.assert_equal(actual,expected)
-
+ assert actual == expected
+
+
def test_error_on_file_to_FileLinks():
"""FileLinks: Raises error when passed file
"""
@@ -178,11 +184,11 @@ def test_recursive_FileLinks():
fl = display.FileLinks(td)
actual = str(fl)
actual = actual.split('\n')
- nt.assert_equal(len(actual), 4, actual)
+ assert len(actual) == 4, actual
fl = display.FileLinks(td, recursive=False)
actual = str(fl)
actual = actual.split('\n')
- nt.assert_equal(len(actual), 2, actual)
+ assert len(actual) == 2, actual
def test_audio_from_file():
path = pjoin(dirname(__file__), 'test.wav')
@@ -194,13 +200,13 @@ class TestAudioDataWithNumpy(TestCase):
def test_audio_from_numpy_array(self):
test_tone = get_test_tone()
audio = display.Audio(test_tone, rate=44100)
- nt.assert_equal(len(read_wav(audio.data)), len(test_tone))
+ assert len(read_wav(audio.data)) == len(test_tone)
@skipif_not_numpy
def test_audio_from_list(self):
test_tone = get_test_tone()
audio = display.Audio(list(test_tone), rate=44100)
- nt.assert_equal(len(read_wav(audio.data)), len(test_tone))
+ assert len(read_wav(audio.data)) == len(test_tone)
@skipif_not_numpy
def test_audio_from_numpy_array_without_rate_raises(self):
@@ -212,7 +218,7 @@ class TestAudioDataWithNumpy(TestCase):
for scale in [1, 0.5, 2]:
audio = display.Audio(get_test_tone(scale), rate=44100)
actual_max_value = numpy.max(numpy.abs(read_wav(audio.data)))
- nt.assert_equal(actual_max_value, expected_max_value)
+ assert actual_max_value == expected_max_value
@skipif_not_numpy
def test_audio_data_without_normalization(self):
@@ -223,7 +229,7 @@ class TestAudioDataWithNumpy(TestCase):
expected_max_value = int(max_int16 * test_tone_max_abs)
audio = display.Audio(test_tone, rate=44100, normalize=False)
actual_max_value = numpy.max(numpy.abs(read_wav(audio.data)))
- nt.assert_equal(actual_max_value, expected_max_value)
+ assert actual_max_value == expected_max_value
def test_audio_data_without_normalization_raises_for_invalid_data(self):
nt.assert_raises(