@@ -51,7 +51,7 @@ def test_valid_weight_shape(data):
5151 missing = PanelData (data .y ).isnull | PanelData (data .x ).isnull
5252 expected = weights [~ missing .squeeze ()][:, None ]
5353 expected = expected / expected .mean ()
54- assert_equal ( w , expected )
54+ assert w == pytest . approx ( expected )
5555
5656 # Per time
5757 if isinstance (data .x , pd .DataFrame ):
@@ -72,7 +72,7 @@ def test_valid_weight_shape(data):
7272 expected = expected .T .ravel ()
7373 expected = expected [~ missing .squeeze ()][:, None ]
7474 expected = expected / expected .mean ()
75- assert_equal ( w , expected )
75+ assert w == pytest . approx ( expected )
7676
7777 # Per entity
7878 if isinstance (data .x , pd .DataFrame ):
@@ -92,7 +92,7 @@ def test_valid_weight_shape(data):
9292 expected = expected .T .ravel ()
9393 expected = expected [~ missing .squeeze ()][:, None ]
9494 expected = expected / expected .mean ()
95- assert_equal ( w , expected )
95+ assert w == pytest . approx ( expected )
9696
9797 weights = 1 + np .random .random_sample (data .y .shape )
9898 mod = PanelOLS (data .y , data .x , weights = weights )
@@ -101,7 +101,7 @@ def test_valid_weight_shape(data):
101101 expected = weights .T .ravel ()
102102 expected = expected [~ missing .squeeze ()][:, None ]
103103 expected = expected / expected .mean ()
104- assert_equal ( w , expected )
104+ assert w == pytest . approx ( expected )
105105
106106
107107def test_weight_incorrect_shape (data ):
0 commit comments