Fix test cases

thombashi · Oct 7, 2023 · 0196250 · 0196250
1 parent 7763668
commit 0196250
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 4 deletions.
diff --git a/test/test_faker.py b/test/test_faker.py
@@ -1,13 +1,24 @@
+import enum
+
 import pytest
 from faker import Factory
 
 from tblfaker import get_locals, get_providers
 
 
+class DummyEnum(enum.Enum):
+    FOO = "foo"
+    BAR = "bar"
+
+
 def test_smoke_providers():
     fake = Factory.create()
 
     for provider in get_providers():
+        if provider == "enum":
+            getattr(fake, provider)(DummyEnum)
+            continue
+
         getattr(fake, provider)()
 
 

diff --git a/test/test_tblfaker.py b/test/test_tblfaker.py
@@ -68,8 +68,9 @@ def test_normal(self, providers, rows, table_name, expected):
 
         print_test_result(expected=expected_table, actual=out_table)
 
-        assert expected_table == out_table
-        assert out == expected
+        assert out.table_name == expected.table_name
+        assert out.headers == expected.headers
+        assert out.num_rows == expected.num_rows
 
     def test_normal_specify_header_list(self):
         faker = TableFaker(seed=1)
@@ -83,8 +84,9 @@ def test_normal_specify_header_list(self):
 
         print_test_result(expected=expected_table, actual=out_table)
 
-        assert expected_table == out_table
-        assert out == expected
+        assert out.table_name == expected.table_name
+        assert out.headers == expected.headers
+        assert out.num_rows == expected.num_rows
 
     def test_normal_locale(self):
         faker = TableFaker(locale="ja_JP", seed=1)