[
https://issues.apache.org/jira/browse/BEAM-9547?focusedWorklogId=610338&page=com.atlassian.jira.plugin.system.issuetabpanels:worklog-tabpanel#worklog-610338
]
ASF GitHub Bot logged work on BEAM-9547:
----------------------------------------
Author: ASF GitHub Bot
Created on: 14/Jun/21 07:40
Start Date: 14/Jun/21 07:40
Worklog Time Spent: 10m
Work Description: robertwb commented on a change in pull request #15002:
URL: https://github.com/apache/beam/pull/15002#discussion_r650242075
##########
File path: sdks/python/apache_beam/dataframe/frames_test.py
##########
@@ -1607,6 +1607,30 @@ def test_sample_with_missing_weights(self):
self.assertEqual(series_result.name, "GDP")
self.assertEqual(set(series_result.index), set(["Nauru", "Iceland"]))
+ def test_sample_with_weights_distribution(self):
+ num_other_elements = 100
+ num_runs = 20
+
+ def sample_many_times(s, weights):
+ all = None
+ for _ in range(num_runs):
+ sampled = s.sample(weights=weights)
+ if all is None:
+ all = sampled
+ else:
+ all = all.append(sampled)
+ return all.sum()
+
+ result = self._run_test(
+ sample_many_times,
+ # The first element is 1, the rest are all 0. This means that when
+ # we sum all the sampled elements (above), the result should be the
+ # number of times the first element was sampled.
+ pd.Series([1] + [0] * num_other_elements),
+ # Pick the first element about 20% of the time.
+ pd.Series([0.2] + [.8 / num_other_elements] * num_other_elements))
+
+ self.assertTrue(0 < result < num_runs / 2, result)
Review comment:
Updated the test, PTAL.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
[email protected]
Issue Time Tracking
-------------------
Worklog Id: (was: 610338)
Time Spent: 119h 10m (was: 119h)
> Implement all pandas operations (or raise WontImplementError)
> -------------------------------------------------------------
>
> Key: BEAM-9547
> URL: https://issues.apache.org/jira/browse/BEAM-9547
> Project: Beam
> Issue Type: Improvement
> Components: sdk-py-core
> Reporter: Brian Hulette
> Assignee: Robert Bradshaw
> Priority: P2
> Labels: dataframe-api
> Time Spent: 119h 10m
> Remaining Estimate: 0h
>
> We should have an implementation for every DataFrame, Series, and GroupBy
> method. Everything that's not possible to implement should get a default
> implementation that raises WontImplementError
> See https://github.com/apache/beam/pull/10757#discussion_r389132292
> Progress at the individual operation level is tracked in a
> [spreadsheet|https://docs.google.com/spreadsheets/d/1hHAaJ0n0k2tw465ORs5tfdy4Lg0DnGWIQ53cLjAhel0/edit],
> consider requesting edit access if you'd like to help out.
--
This message was sent by Atlassian Jira
(v8.3.4#803005)
