Add test with assertRaises

Author: suaaa7

python_ci/src/tests/test_df_service.py

@@ -99,6 +99,19 @@ def test_return_expected_df_when_get_local_path_return_none(self) -> None:
         assert_frame_equal(actual_df, expected_df)
         self.assertEqual(self.m_get_local_path.call_count, 3)
 
+    def test_raise_attribute_error_when_no_image_hash(self) -> None:
+        self.m_get_local_path.return_value = '/opt/program/test_image.jpg'
+
+        bad_input_df = pd.DataFrame({
+            'num': [1],
+            'image_path': [
+                ['/opt/program/test_image.jpg'],
+            ]
+        })
+
+        with self.assertRaises(AttributeError):
+            self.service.swap_and_remove_path(bad_input_df)
+
     @skip('Use unittest.skip')
     def test_tmp(self) -> None:
         pass