black it

snorkel-team · Jun 21, 2020 · 24de6cb · 24de6cb
1 parent bc52960
commit 24de6cb
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/test/map/test_core.py b/test/map/test_core.py
@@ -271,12 +271,16 @@ def square(x: DataPoint) -> DataPoint:
             x.num_squared = square_hit_tracker(x.num)
             return x
 
-        x1 = SimpleNamespace(uid="id1", num=8, unhashable=pd.DataFrame({"value": [5]}))
+        x1 = SimpleNamespace(
+            uid="id1", num=8, not_used=0, unhashable=pd.DataFrame({"value": [5]})
+        )
         x1_mapped = square(x1)
         assert x1_mapped is not None
         self.assertEqual(x1_mapped.num_squared, 64)
         self.assertEqual(square_hit_tracker.n_hits, 1)
-        x2 = SimpleNamespace(uid="id1", num=8, unhashable=pd.DataFrame({"value": [5]}))
+        x2 = SimpleNamespace(
+            uid="id1", num=8, not_used=1, unhashable=pd.DataFrame({"value": [5]})
+        )
         x2_mapped = square(x2)
         assert x2_mapped is not None
         self.assertEqual(x2_mapped.num_squared, 64)