|
4 | 4 |
|
5 | 5 | from flask import Flask, request, abort, url_for |
6 | 6 | from flask.views import MethodView |
| 7 | + |
7 | 8 | from flask_admin import base |
8 | 9 |
|
9 | 10 |
|
@@ -233,23 +234,23 @@ def test_add_category(): |
233 | 234 | admin.add_view(MockView(name='Test 1', endpoint='test1', category='Category1')) |
234 | 235 | admin.add_view(MockView(name='Test 2', endpoint='test2', category='Category2')) |
235 | 236 |
|
236 | | - eq_(len(admin.menu()), 3) |
| 237 | + assert len(admin.menu()) == 3 |
237 | 238 |
|
238 | 239 | # Test 1 should be underneath Category1 |
239 | | - eq_(admin.menu()[1].name, 'Category1') |
240 | | - eq_(admin.menu()[1].get_class_name(), 'class-name') |
241 | | - eq_(admin.menu()[1].get_icon_type(), 'icon-type') |
242 | | - eq_(admin.menu()[1].get_icon_value(), 'icon-value') |
243 | | - eq_(len(admin.menu()[1].get_children()), 1) |
244 | | - eq_(admin.menu()[1].get_children()[0].name, 'Test 1') |
| 240 | + assert admin.menu()[1].name == 'Category1' |
| 241 | + assert admin.menu()[1].get_class_name() == 'class-name' |
| 242 | + assert admin.menu()[1].get_icon_type() == 'icon-type' |
| 243 | + assert admin.menu()[1].get_icon_value() == 'icon-value' |
| 244 | + assert len(admin.menu()[1].get_children()) == 1 |
| 245 | + assert admin.menu()[1].get_children()[0].name == 'Test 1' |
245 | 246 |
|
246 | 247 | # Test 2 should be underneath Category2 |
247 | | - eq_(admin.menu()[2].name, 'Category2') |
248 | | - eq_(admin.menu()[2].get_class_name(), None) |
249 | | - eq_(admin.menu()[2].get_icon_type(), None) |
250 | | - eq_(admin.menu()[2].get_icon_value(), None) |
251 | | - eq_(len(admin.menu()[2].get_children()), 1) |
252 | | - eq_(admin.menu()[2].get_children()[0].name, 'Test 2') |
| 248 | + assert admin.menu()[2].name == 'Category2' |
| 249 | + assert admin.menu()[2].get_class_name() == None |
| 250 | + assert admin.menu()[2].get_icon_type() == None |
| 251 | + assert admin.menu()[2].get_icon_value() == None |
| 252 | + assert len(admin.menu()[2].get_children()) == 1 |
| 253 | + assert admin.menu()[2].get_children()[0].name == 'Test 2' |
253 | 254 |
|
254 | 255 |
|
255 | 256 | @pytest.mark.xfail(raises=Exception) |
|
0 commit comments