Karim shoair commited on
Commit
3e4349d
·
1 Parent(s): 243d94d

tests: change tests accordingly

Browse files
Files changed (1) hide show
  1. tests/cli/test_shell_functionality.py +14 -16
tests/cli/test_shell_functionality.py CHANGED
@@ -178,23 +178,21 @@ class TestCustomShell:
178
 
179
  def test_shell_initialization(self):
180
  """Test shell initialization"""
181
- with patch('scrapling.core.shell.InteractiveShellEmbed'):
182
- shell = CustomShell(code="", log_level="debug")
183
 
184
- assert shell.log_level == 10 # DEBUG level
185
- assert shell.page is None
186
- assert len(shell.pages) == 0
187
 
188
  def test_shell_namespace(self):
189
  """Test shell namespace creation"""
190
- with patch('scrapling.core.shell.InteractiveShellEmbed'):
191
- shell = CustomShell(code="")
192
- namespace = shell.get_namespace()
193
-
194
- # Check all expected functions/classes are available
195
- assert 'get' in namespace
196
- assert 'post' in namespace
197
- assert 'Fetcher' in namespace
198
- assert 'DynamicFetcher' in namespace
199
- assert 'view' in namespace
200
- assert 'uncurl' in namespace
 
178
 
179
  def test_shell_initialization(self):
180
  """Test shell initialization"""
181
+ shell = CustomShell(code="", log_level="debug")
 
182
 
183
+ assert shell.log_level == 10 # DEBUG level
184
+ assert shell.page is None
185
+ assert len(shell.pages) == 0
186
 
187
  def test_shell_namespace(self):
188
  """Test shell namespace creation"""
189
+ shell = CustomShell(code="")
190
+ namespace = shell.get_namespace()
191
+
192
+ # Check all expected functions/classes are available
193
+ assert 'get' in namespace
194
+ assert 'post' in namespace
195
+ assert 'Fetcher' in namespace
196
+ assert 'DynamicFetcher' in namespace
197
+ assert 'view' in namespace
198
+ assert 'uncurl' in namespace