Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Changing the name for the new statisticaltest resource
- Loading branch information
Showing
12 changed files
with
309 additions
and
265 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,120 @@ | ||
| # -*- coding: utf-8 -*- | ||
| #!/usr/bin/env python | ||
| # | ||
| # Copyright 2015 BigML | ||
| # | ||
| # Licensed under the Apache License, Version 2.0 (the "License"); you may | ||
| # not use this file except in compliance with the License. You may obtain | ||
| # a copy of the License at | ||
| # | ||
| # http://www.apache.org/licenses/LICENSE-2.0 | ||
| # | ||
| # Unless required by applicable law or agreed to in writing, software | ||
| # distributed under the License is distributed on an "AS IS" BASIS, WITHOUT | ||
| # WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the | ||
| # License for the specific language governing permissions and limitations | ||
| # under the License. | ||
|
|
||
| """Base class for statisticaltests' REST calls | ||
| https://bigml.com/developers/statisticaltests | ||
| """ | ||
|
|
||
| try: | ||
| import simplejson as json | ||
| except ImportError: | ||
| import json | ||
|
|
||
|
|
||
| from bigml.resourcehandler import ResourceHandler | ||
| from bigml.resourcehandler import (check_resource_type, | ||
| get_statistical_test_id, get_resource_type, | ||
| get_dataset_id, check_resource) | ||
| from bigml.resourcehandler import (STATISTICAL_TEST_PATH, DATASET_PATH, | ||
| TINY_RESOURCE) | ||
|
|
||
|
|
||
| class StatisticalTestHandler(ResourceHandler): | ||
| """This class is used by the BigML class as | ||
| a mixin that provides the statistical tests' REST calls. It should not | ||
| be instantiated independently. | ||
| """ | ||
| def __init__(self): | ||
| """Initializes the StatisticalTestHandler. This class is intended to be | ||
| used as a mixin on ResourceHandler, that inherits its | ||
| attributes and basic method from BigMLConnection, and must not be | ||
| instantiated independently. | ||
| """ | ||
| self.statistical_test_url = self.url + STATISTICAL_TEST_PATH | ||
|
|
||
| def create_statistical_test(self, dataset, args=None, wait_time=3, retries=10): | ||
| """Creates a statistical test from a `dataset`. | ||
| """ | ||
| dataset_id = None | ||
| resource_type = get_resource_type(dataset) | ||
| if resource_type == DATASET_PATH: | ||
| dataset_id = get_dataset_id(dataset) | ||
| check_resource(dataset_id, | ||
| query_string=TINY_RESOURCE, | ||
| wait_time=wait_time, retries=retries, | ||
| raise_on_error=True, api=self) | ||
| else: | ||
| raise Exception("A dataset id is needed to create a" | ||
| " statistical test. %s found." % resource_type) | ||
|
|
||
| create_args = {} | ||
| if args is not None: | ||
| create_args.update(args) | ||
| create_args.update({ | ||
| "dataset": dataset_id}) | ||
|
|
||
| body = json.dumps(create_args) | ||
| return self._create(self.statistical_test_url, body) | ||
|
|
||
| def get_statistical_test(self, statistical_test, query_string=''): | ||
| """Retrieves a statistical test. | ||
| The statistical test parameter should be a string containing the | ||
| statisticaltest id or the dict returned by create_statistical_test. | ||
| As an statistical test is an evolving object that is processed | ||
| until it reaches the FINISHED or FAULTY state, the function will | ||
| return a dict that encloses the statistical test values and state | ||
| info available at the time it is called. | ||
| """ | ||
| check_resource_type(statistical_test, STATISTICAL_TEST_PATH, | ||
| message="A statistical test id is needed.") | ||
| statistical_test_id = get_statistical_test_id(statistical_test) | ||
| if statistical_test_id: | ||
| return self._get("%s%s" % (self.url, statistical_test_id), | ||
| query_string=query_string) | ||
|
|
||
| def list_statistical_tests(self, query_string=''): | ||
| """Lists all your statistical tests. | ||
| """ | ||
| return self._list(self.statistical_test_url, query_string) | ||
|
|
||
| def update_statistical_test(self, statistical_test, changes): | ||
| """Updates an statistical test. | ||
| """ | ||
| check_resource_type(statistical_test, STATISTICAL_TEST_PATH, | ||
| message="A statistical test id is needed.") | ||
| statistical_test_id = get_statistical_test_id(statistical_test) | ||
| if statistical_test_id: | ||
| body = json.dumps(changes) | ||
| return self._update("%s%s" % (self.url, statistical_test_id), body) | ||
|
|
||
| def delete_statistical_test(self, statistical_test): | ||
| """Deletes a statistical test. | ||
| """ | ||
| check_resource_type(statistical_test, STATISTICAL_TEST_PATH, | ||
| message="A statistical test id is needed.") | ||
| statistical_test_id = get_statistical_test_id(statistical_test) | ||
| if statistical_test_id: | ||
| return self._delete("%s%s" % (self.url, statistical_test_id)) |
Oops, something went wrong.